-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
106 lines (94 loc) · 1.49 KB
/
main.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package main
import "fmt"
/**
* 合并两个有序链表
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
type ListNode struct {
Val int
Next *ListNode
}
func mergeTwoLists(l1 *ListNode, l2 *ListNode) *ListNode {
if l1 == nil {
return l2
}
if l2 == nil {
return l1
}
// 声明一下新的节点
var newNode *ListNode
// 声明新节点的前置结点
var preNode *ListNode
// 声明一个头结点
head := &ListNode{
Val: 0,
Next: nil,
}
var val int
for l1 != nil && l2 != nil {
if l1.Val < l2.Val {
val = l1.Val
l1 = l1.Next
} else {
val = l2.Val
l2 = l2.Next
}
newNode = &ListNode{
Val: val,
Next: nil,
}
// 首节点的Next是否为空
if head.Next == nil {
preNode = newNode
head.Next = newNode
} else {
preNode.Next = newNode
preNode = newNode
}
}
// 如果 l1 有剩余则直接接后面即可
if l1 != nil {
preNode.Next = l1
}
// 如果 l2 有剩余则直接接后面即可
if l2 != nil {
preNode.Next = l2
}
return head.Next
}
func main() {
l12 := &ListNode{
Val: 4,
Next: nil,
}
l11 := &ListNode{
Val: 2,
Next: l12,
}
l1 := &ListNode{
Val: 1,
Next: l11,
}
l22 := &ListNode{
Val: 4,
Next: nil,
}
l21 := &ListNode{
Val: 3,
Next: l22,
}
l2 := &ListNode{
Val: 1,
Next: l21,
}
mergeLink := mergeTwoLists(l1, l2)
//fmt.Println(mergeLink)
for mergeLink != nil {
fmt.Println(mergeLink)
mergeLink = mergeLink.Next
}
}