Golang刷題-2. 兩數相加

1. 題目描述【中等】

給出兩個 非空 的鏈表用來表示兩個非負的整數。其中，它們各自的位數是按照 逆序 的方式存儲的，並且它們的每個節點只能存儲 一位 數字。
如果，我們將這兩個數相加起來，則會返回一個新的鏈表來表示它們的和。
您可以假設除了數字 0 之外，這兩個數都不會以 0 開頭。

示例：

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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/add-two-numbers
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

2. 解題思路

本題並不算難，思路也很清晰，只不過是對鏈表的一些操作細節需要注意。

2.1 直接對鏈表進行操作動畫參考

注：圖片來自五分鐘學算法公衆號

算法流程：

1. 首先對給定鏈表進行非空判斷，並定義保存結果的head鏈表
2. 對l1與l2同時進行遍歷，若l1或l2已經爲空則不再遍歷該鏈表
3. 將得到的數字相加結果保存爲sum
4. 如果sum>=10則代表需要進位，則用flag作爲標記，同時將sum=sum-10
5. 否則不需要進位，將flag標記爲0
6. 採用尾插法將結果保存在head鏈表中

3. 代碼實現

func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
	if l1 == nil && l2 == nil {
		return nil
	}else if l1 == nil{
		return l2
	}else if l2 == nil{
		return l1
	}
	head := &ListNode{
		Val:-1,
		Next:nil,
	}
	tail := head

	flag := 0
	for l1 != nil || l2 != nil{
		val1, val2 := 0, 0
		if l1 != nil{
			val1 = l1.Val
			l1 = l1.Next
		}
		if l2 != nil{
			val2 = l2.Val
			l2 = l2.Next
		}

		sum := val1 + val2 + flag
		if sum >= 10 {
			flag = 1
			sum -= 10
		}else{
			flag = 0
		}
		tail = listAdd(tail, sum)
	}
	if flag == 1{
		tail = listAdd(tail, 1)
	}
	return head.Next
}

//根據題目要求採用尾插法創建鏈表
func listAdd(tail *ListNode, num int)*ListNode{
	numNode := &ListNode{
		Val:num,
		Next:nil,
	}
	tail.Next  = numNode
	tail = numNode
	return tail
}