2.两数相加(Swift版)

一、题目

给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。
你可以假设除了数字 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

    原文作者:ma772528138
    原文地址: https://www.jianshu.com/p/c2dfb6fa26ef
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞