程式設計題-兩數相加

紅牛慕課_韓忠康發表於2019-11-04

兩數相加

題目

來自: leetcode-cn.com/problems/ad…

給出兩個 非空 的連結串列用來表示兩個非負的整數。其中,它們各自的位數是按照逆序的方式儲存的,並且它們的每個節點只能儲存一位數字。

如果,我們將這兩個數相加起來,則會返回一個新的連結串列來表示它們的和。

您可以假設除了數字 0 之外,這兩個數都不會以 0 開頭。

示例:

輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出:7 -> 0 -> 8
原因:342 + 465 = 807
複製程式碼

Go程式設計實現

思路:迭代計算每位數字的和,記錄進位。當全部數位和進位計算完畢,結果得到。

程式碼如下:

// 單向連結串列結構
type ListNode struct {
	Val  int
	Next *ListNode
}

func AddTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
	//判斷是否存在運算元為 0 的特例
	if l1.Val == 0 && l1.Next == nil {
		return l2
	}
	if l2.Val == 0 && l2.Next == nil {
		return l1
	}

	result := &ListNode{}
	//迭代處理的結果節點,從第一個開始
	node := result
	// 進位
	var carry int
	// 和
	v := 0
	for {
		// 若 l1 還有,則累加,同時切換下一個l1
		if l1 != nil {
			v += l1.Val
			l1 = l1.Next
		}
		// 若 l2 還有,則累加,同時切換下一個 l2
		if l2 != nil {
			v += l2.Val
			l2 = l2.Next
		}
		//累加進位
		v += carry
		carry = v / 10 // 進位
		node.Val = v % 10 // 當前位
		// 若 l1 l2 進位都沒了,結束
		if l1 == nil && l2 == nil && carry == 0 {
			break
		}
		// 構建下一個結果節點
		node.Next = &ListNode{}
		node = node.Next
		v = 0
	}

	return result
}
複製程式碼

本實現在 LeetCode 上測試通過,結果為:

執行用時 : 8 ms , 在所有 golang 提交中擊敗了 96.84% 的使用者。

記憶體消耗 : 4.9 MB , 在所有 golang 提交中擊敗了 94.01% 的使用者。

關注 紅牛慕課,或進入 紅牛慕課 獲取更多程式設計題。

程式設計題-兩數相加

相關文章