1. 题目描述【中等】
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题思路
本题并不算难,思路也很清晰,只不过是对链表的一些操作细节需要注意。
2.1 直接对链表进行操作动画参考
注:图片来自五分钟学算法公众号
算法流程:
- 首先对给定链表进行非空判断,并定义保存结果的
head
链表 - 对
l1
与l2
同时进行遍历,若l1
或l2
已经为空则不再遍历该链表 - 将得到的数字相加结果保存为
sum
- 如果
sum>=10
则代表需要进位,则用flag
作为标记,同时将sum=sum-10
- 否则不需要进位,将
flag
标记为0
- 采用尾插法将结果保存在
head
链表中
3. 代码实现
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
if l1 == nil && l2 == nil {
return nil
}else if l1 == nil{
return l2
}else if l2 == nil{
return l1
}
head := &ListNode{
Val:-1,
Next:nil,
}
tail := head
flag := 0
for l1 != nil || l2 != nil{
val1, val2 := 0, 0
if l1 != nil{
val1 = l1.Val
l1 = l1.Next
}
if l2 != nil{
val2 = l2.Val
l2 = l2.Next
}
sum := val1 + val2 + flag
if sum >= 10 {
flag = 1
sum -= 10
}else{
flag = 0
}
tail = listAdd(tail, sum)
}
if flag == 1{
tail = listAdd(tail, 1)
}
return head.Next
}
//根据题目要求采用尾插法创建链表
func listAdd(tail *ListNode, num int)*ListNode{
numNode := &ListNode{
Val:num,
Next:nil,
}
tail.Next = numNode
tail = numNode
return tail
}