一、题目
给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。
你可以假设除了数字 0 之外,这两个数字都不会以零开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
二、解题
通过示例,可以看出规律,链表的表头就是第一位数,将两个链表分别的命名为a和b,输入链表为c,a1+b1=c1,同时,如果c1>=10,需要向后进位,c1仅保留个数。其中需要注意的是两个链表的长度可能会出现不一致的情况。
三、代码实现
/**
* Definition for singly-linked list.
* public class ListNode {
* public var val: Int
* public var next: ListNode?
* public init(_ val: Int) {
* self.val = val
* self.next = nil
* }
* }
*/
class Solution {
func addTwoNumbers(_ l1: ListNode?, _ l2: ListNode?) -> ListNode? {
// 定义输出节点c
var c: ListNode? = nil
// 进位
var carry = 0
// 循环使用的链表节点
var n:ListNode? = nil
var a = l1
var b = l2
// a.next或b.next或carry存在,均需要继续增加节点,用于处理两个链表长度可能不一致的问题
while (a != nil || b != nil || carry > 0) {
let count : Int = (a?.val ?? 0) + (b?.val ?? 0) + carry
// 获取进位
carry = count / 10
// 如果c不存在,则创建表头,如果c存在,则设置n.next,向后移动
if c == nil {
c = ListNode.init(count % 10)
n = c
}else{
n?.next = ListNode.init(count % 10)
n = n?.next
}
a = a?.next
b = b?.next
}
return c
}
}
Demo地址:github