题目
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
方案一
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
dummyHead := &ListNode {}
node := dummyHead
var carry int // 表示进位的数字
for l1 != nil || l2 != nil {
var newNode ListNode
if l1 == nil {
newNode.Val = (carry + l2.Val) % 10
carry = (carry + l2.Val) / 10
l2 = l2.Next
node.Next = &newNode // 注意:不能提取到最后 7->3 + 0
node = &newNode
continue
}
if l2 == nil {
newNode.Val = (carry + l1.Val) % 10
carry = (carry + l1.Val) / 10
l1 = l1.Next
node.Next = &newNode // 注意:不能提取到最后
node = &newNode
continue
}
if l1 != nil && l2 != nil {
newNode.Val = (carry + l1.Val + l2.Val) % 10
carry = (carry + l1.Val + l2.Val) / 10
l1 = l1.Next
l2 = l2.Next
node.Next = &newNode // 注意:不能提取到最后
node = &newNode
}
}
// 处理最后一个进位
if carry != 0 {
node.Next = &ListNode {
Val: carry,
}
}
return dummyHead.Next
}
- 根据加法规则,生成一个新的链表并返回。
优化后的代码如下:
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
dummyHead := &ListNode {}
node := dummyHead
var carry int // 表示进位的数字
for l1 != nil || l2 != nil {
if l1 != nil {
carry += l1.Val
l1 = l1.Next
}
if l2 != nil {
carry += l2.Val
l2 = l2.Next
}
node.Next = &ListNode {
Val: carry % 10,
}
carry = carry / 10
node = node.Next
}
// 处理最后一个进位
if carry != 0 {
node.Next = &ListNode {Val: carry}
}
return dummyHead.Next
}
原文
https://leetcode-cn.com/explore/learn/card/linked-list/197/conclusion/763/