【leetcode--golang】2.两数相加

题目

给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。您可以假设除了数字 0 之外，这两个数都不会以 0 开头。

示例

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

解决方案

就像在纸上计算两个数字的和那样，我们首先从最低有效位也就是列表 l1 和 l2 的表头开始相加。由于每位数字都应当处于 0…9 的范围内，我们计算两个数字的和时可能会出现“溢出”。例如，5 + 7 = 12。在这种情况下，我们会将当前位的数值设置为 2，并将进位 flag = 1 带入下一次迭代。进位 flag 必定是 0 或 1，这是因为两个数字相加（考虑到进位）可能出现的最大和为 9 + 9 + 1 = 19。

func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {	var value, flag int
	var l3, cur *ListNode
	if l1 == nil && l2 == nil {
		return initListNode(0)
	}

	for {
		sum := 0
		if l1 == nil && l2 == nil {
			if flag != 0 {
				cur.Next = initListNode(flag)
			}
			return l3
		}

		if l1 != nil {
			sum += l1.Val
			l1 = l1.Next
		}

		if l2 != nil {
			sum += l2.Val
			l2 = l2.Next
		}

		if flag != 0 {
			sum += flag
		}

		value = sum % 10
		flag = sum / 10

		if l3 != nil {
			cur.Next = initListNode(value)
			cur = cur.Next
		} else {
			l3 = initListNode(value)
			cur = l3
		}
	}
}

func initListNode(n int) *ListNode {
	node := &ListNode{}
	node.Val = n
	node.Next = nil
	return node
}

复杂度分析

• 时间复杂度：O(max(m, n))，假设 m 和 n 分别表示 l1l1 和 l2 的长度，上面的算法最多重复 max(m, n) 次。
• 空间复杂度：O(max(m, n))， 新列表的长度最多为 max(m,n) + 1max(m,n)+1。