LeetCode-0002 两数相加
作者:互联网
题目
给出两个【非空】的链表用来表示两个非负的整数。其中,它们各自的位数是按照【逆序】的方式存储的,并且它们的每个节点只能存储【一位】数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
说明
链表的节点按如下定义
type ListNode struct {
Val int
Next *ListNode
}
解答
注意处理进位。
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
var (
t1 *ListNode = l1
t2 *ListNode = l2
p *ListNode
)
// 为编写代码方便,设立一个哑节点
ret := &ListNode{
Val: 0,
Next: nil,
}
p = ret
var (
x int
y int
carry int = 0
s int = 0
)
for t1 != nil || t2 != nil {
if t1 != nil {
x = t1.Val
} else {
x = 0
}
if t2 != nil {
y = t2.Val
} else {
y = 0
}
s = x + y + carry
carry = s / 10
p.Next = &ListNode{
Val: s % 10,
Next: nil,
}
p = p.Next
if t1 != nil {
t1 = t1.Next
}
if t2 != nil {
t2 = t2.Next
}
}
if carry != 0 {
p.Next = &ListNode{
Val: carry,
Next: nil,
}
}
// 把哑节点去掉再返回
ret = ret.Next
return ret
}
用来创建测试用数据的函数
func createTestData(a []int, b []int) (*ListNode, *ListNode) {
const textErr = "bad param"
if a == nil || b == nil {
panic(textErr)
}
m, n := len(a), len(b)
if m == 0 || n == 0 {
panic(textErr)
}
if (a[0] == 0 && m != 1) || (b[0] == 0 && n != 1) {
panic(textErr)
}
for i := 0; i < m; i++ {
if a[i] < 0 || a[i] > 9 {
panic(textErr)
}
}
for i := 0; i < n; i++ {
if b[i] < 0 || b[i] > 9 {
panic(textErr)
}
}
var l1 *ListNode
var l2 *ListNode
var p *ListNode
var q *ListNode
for i := 0; i < m; i++ {
p = &ListNode{
Val: a[i],
Next: nil,
}
if i == 0 {
q = p
l1 = p
} else {
q.Next = p
q = p
}
}
for i := 0; i < n; i++ {
p = &ListNode{
Val: b[i],
Next: nil,
}
if i == 0 {
q = p
l2 = p
} else {
q.Next = p
q = p
}
}
return l1, l2
}
标签:ListNode,0002,nil,相加,t2,Next,Val,t1,LeetCode 来源: https://www.cnblogs.com/jthmath/p/12124897.html