两数相加 golang


2.两数相加[中等]

题干

给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。

请你将两个数相加,并以相同形式返回一个表示和的链表。

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

示例 1:

img

输入:l1 = [2,4,3], l2 = [5,6,4]
输出:[7,0,8]
解释:342 + 465 = 807.

示例2:

输入:l1 = [0], l2 = [0]
输出:[0]

示例3:

输入:l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出:[8,9,9,9,0,0,0,1]

题解

思路:

  1. 两个链表非空则向下遍历,Val相加。
  2. 记录进位,再每轮计算的值中加上进位。
  3. 创建下一个结构,存入计算的值,向下遍历。

题解:

func addTwoNumbers(l1, l2 *ListNode) (head *ListNode) {
    //操作结果链表的指针
    var tail *ListNode
    //进位值
    carry := 0
    //两个链表有其一非空则遍历
    for l1 != nil || l2 != nil {
        n1, n2 := 0, 0
  		//链表非空则遍历下一个值
        if l1 != nil {
            n1 = l1.Val
            l1 = l1.Next
        }
        if l2 != nil {
            n2 = l2.Val
            l2 = l2.Next
        }
        //计算当前节点值
        sum := n1 + n2 + carry
        sum, carry = sum%10, sum/10
        //针对首次遍历则创建首个结点
        if head == nil {
            head = &ListNode{Val: sum}
            tail = head
        } else {
         //针对非首次遍历则创建下个结点并继续遍历
            tail.Next = &ListNode{Val: sum}
            tail = tail.Next
        }
    }
    //考虑最后遍历完l1、l2后仍有进位
    if carry > 0 {
        tail.Next = &ListNode{Val: carry}
    }
    return
}