刷题--程序员面试金典--面试题 02.05. 链表求和(重点)（go）

面试题 02.05. 链表求和

给定两个用链表表示的整数，每个节点包含一个数位。

这些数位是反向存放的，也就是个位排在链表首部。

编写函数对这两个整数求和，并用链表形式返回结果。

 

示例：

输入：(7 -> 1 -> 6) + (5 -> 9 -> 2)，即617 + 295
输出：2 -> 1 -> 9，即912
进阶：假设这些数位是正向存放的，请再做一遍。

示例：

输入：(6 -> 1 -> 7) + (2 -> 9 -> 5)，即617 + 295
输出：9 -> 1 -> 2，即912

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-lists-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

go实现：

func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
    resHead := &ListNode{0,nil}
    p := resHead
    flag := 0


    for l1 != nil && l2 != nil {
        sum := l1.Val + l2.Val + flag
        t := &ListNode{0,nil}
        if sum >= 10 {
            t.Val = sum%10
            flag = 1
        } else {
            flag = 0
            t.Val = sum
        }
        p.Next = t
        l1 = l1.Next
        l2 = l2.Next
        p = p.Next
    }

    for l1 != nil {
        sum := l1.Val + flag
        t := &ListNode{0,nil}
        if sum >= 10 {
            t.Val = sum%10
            flag = 1
        } else {
            flag = 0
            t.Val = sum
        }
        p.Next = t
        l1 = l1.Next
        p = p.Next
    }

    for l2 != nil {
        sum := l2.Val + flag
        t := &ListNode{0,nil}
        if sum >= 10 {
            t.Val = sum%10
            flag = 1
        } else {
            flag = 0
            t.Val = sum
        }
        p.Next = t
        l2 = l2.Next
        p = p.Next
    }

    if flag == 1 {
        t := &ListNode{1,nil}
        p.Next = t
    }

    return resHead.Next
}

代码优化：

func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
    resHead := &ListNode{}
    p := resHead
    sum := 0

    for l1 != nil || l2 != nil || sum != 0 {
        if l1 != nil {
            sum += l1.Val
            l1 = l1.Next
        }
        if l2 != nil {
            sum += l2.Val
            l2 = l2.Next
        }
        p.Next = &ListNode{sum%10,nil}

        p = p.Next
        sum /= 10
    }

    return resHead.Next
}