LeetCode 369. 给单链表加一
题目描述
思路分析
加法问题
参考代码
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
54
55
56
57
58
59
60
61
62
63
// ListNode 定义单链表节点
type ListNode struct {
Val int
Next *ListNode
}
// plusOne 给单链表加一
func plusOne(head *ListNode) *ListNode {
dummy := &ListNode{Next: head}
notNine := dummy
// 找到最后一个不是 9 的节点
for head != nil {
if head.Val != 9 {
notNine = head
}
head = head.Next
}
// 将最后一个不是 9 的节点加一
notNine.Val++
notNine = notNine.Next
// 将后续的所有 9 变成 0
for notNine != nil {
notNine.Val = 0
notNine = notNine.Next
}
// 如果 dummy 的值被改变，说明有进位，需要返回 dummy，否则返回原链表头。
if dummy.Val == 1 {
return dummy
}
return dummy.Next
}
// 打印链表
func printList(head *ListNode) {
for head != nil {
fmt.Print(head.Val)
if head.Next != nil {
fmt.Print(" -> ")
}
head = head.Next
}
fmt.Println()
}
func main() {
// 创建一个示例链表 1 -> 2 -> 3
head := &ListNode{Val: 9}
head.Next = &ListNode{Val: 9}
head.Next.Next = &ListNode{Val: 9}
fmt.Print("原链表: ")
printList(head)
// 给链表加一
head = plusOne(head)
fmt.Print("加一后的链表: ")
printList(head)
}
原链表: 1 -> 2 -> 3 加一后的链表: 1 -> 2 -> 4
原链表: 9 -> 9 -> 9 加一后的链表: 1 -> 0 -> 0 -> 0
1
write your code here
本博客所有文章除特别声明外，均采用
CC BY-NC-SA 4.0
许可协议，转载请注明出处！