forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSolution.go
53 lines (51 loc) · 1.36 KB
/
Solution.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*
* Report by leetcode.com
* Runtime: 12 ms, Memory Usage: 5 MB
*/
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
head := &ListNode{}
currentNode1 := l1
currentNode2 := l2
currentHead := head
sum := 0
nextSum := 0
for true {
if currentNode1 != nil || currentNode2 != nil {
if currentNode1 == nil {
sum = nextSum + currentNode2.Val
nextSum = sum / 10
currentNode2 = currentNode2.Next
} else if currentNode2 == nil {
sum = nextSum + currentNode1.Val
nextSum = sum / 10
currentNode1 = currentNode1.Next
} else {
sum = nextSum + currentNode1.Val + currentNode2.Val
nextSum = sum / 10
currentNode1 = currentNode1.Next
currentNode2 = currentNode2.Next
}
currentHead.Val = sum % 10
// If there are elements present in the nodes then
// make a new node for future addition otherwise we
// will get unnecessary (0 --> <nil>) node in the end.
if currentNode1 != nil || currentNode2 != nil {
currentHead.Next = &ListNode{}
currentHead = currentHead.Next
} else if nextSum != 0 {
// If nextSum is not 0 this means that there was some carry value
// left in it which should be further.
currentHead.Next = &ListNode{nextSum, nil}
}
} else {
break
}
}
return head
}