leetcode 经典面试题之链表 (2. 两数相加)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_36431213/article/details/84262032
  • 两数相加

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 单位 数字。
如果,我们将这两个数起来相加起来,则会返回出一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807

func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
	res := new(ListNode)
	cur := res
	carry := false

	for l1 != nil || l2 != nil {
		sum := 0
		if l1 != nil {
			sum = sum + l1.Val
			l1 = l1.Next
		}
		if l2 != nil {
			sum = sum + l2.Val
			l2 = l2.Next
		}

		if carry {
			sum++
		}

		if sum > 9 {
			sum = sum - 10
			carry = true
		} else {
			carry = false
		}
		cur.Next = &ListNode{Val: sum}
		cur = cur.Next
	}

	if carry {
		cur.Next = &ListNode{Val: 1}
	}
	return res.Next
}

猜你喜欢

转载自blog.csdn.net/qq_36431213/article/details/84262032