[leetcode]两数相加

空闲时间用Go刷了一道水题, 题目和题解如下:

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。

如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
    // 进位(满十进一)
    carry := 0
    // 当次运算的和
    sum := 0
    // 结果链表中的表头节点
    var head *ListNode
    // 结果链表中的上一次使用的节点
    var last *ListNode
    // 链表1中当前访问的节点, 初始值为表头节点
    node1 := l1
    // 链表2中当前访问的节点, 初始值为表头节点
    node2 := l2

    // 1. 完成两个链表共同低位的运算
    for node1!=nil && node2!=nil {
        newNode := &ListNode{
            Val : 0,
            Next: nil,
        }

        sum = node1.Val + node2.Val + carry
        if (sum >= 10) {
            carry = 1
            newNode.Val = sum - 10
        } else {
            carry = 0
            newNode.Val = sum
        }

        if (head == nil) {
            head = newNode
            last = head
        } else {
            last.Next = newNode
            last = newNode
        }

        node1 = node1.Next
        node2 = node2.Next
    }

    // 2. 如果两个链表长度不一样, 完成较长链表剩余高位的运算
    node3 := node1
    if node3 == nil {
        node3 = node2
    }
    for node3!=nil {
        if (carry == 0) {
            last.Next = node3
            last = node3
            node3 = node3.Next
            continue
        }

        newNode := &ListNode{
            Val : 0,
            Next: nil,
        }

        sum = node3.Val + carry
        if (sum >= 10) {
            carry = 1
            newNode.Val = sum - 10
        } else {
            carry = 0
            newNode.Val = sum
        }

        last.Next = newNode
        last = newNode
        node3 = node3.Next
    }

    // 3. 最后一次处理进位
    if carry > 0 {
        newNode := &ListNode{
            Val : carry,
            Next: nil,
        }
        last.Next = newNode
        last = newNode
        carry = 0
    }

    return head
}
发布了372 篇原创文章 · 获赞 1031 · 访问量 242万+

猜你喜欢

转载自blog.csdn.net/moxiaomomo/article/details/103796726