这是链表的第4题,来个简单算法玩玩。力扣链接。
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例 1:
输入:l1 = [1,2,4], l2 = [1,3,4] 输出:[1,1,2,3,4,4]示例 2:
输入:l1 = [], l2 = [] 输出:[]示例 3:
输入:l1 = [], l2 = [0] 输出:[0]
看到这个题不假思索的就会考虑双指针解题。
func mergeTwoLists(list1 *ListNode, list2 *ListNode) *ListNode {
result := &ListNode{}
cur := result
for list1 != nil || list2 != nil {
left := math.MaxInt
if list1 != nil {
left = list1.Val
}
right := math.MaxInt
if list2 != nil {
right = list2.Val
}
if left < right {
cur.Next = &ListNode{
Val: left,
}
list1 = list1.Next
} else {
cur.Next = &ListNode{
Val: right,
}
list2 = list2.Next
}
cur = cur.Next
}
return result.Next
}
得出的结果性能很差,反思一下,如果一边链表为空了,还有必要继续遍历吗,答案是没有必要。
优化如下:
func mergeTwoLists(list1 *ListNode, list2 *ListNode) *ListNode {
result := &ListNode{}
cur := result
for list1 != nil && list2 != nil {
if list1.Val < list2.Val {
cur.Next = &ListNode{
Val: list1.Val,
}
list1 = list1.Next
} else {
cur.Next = &ListNode{
Val: list2.Val,
}
list2 = list2.Next
}
cur = cur.Next
}
if list1 != nil {
cur.Next = list1
}
if list2 != nil {
cur.Next = list2
}
return result.Next
}
当然也可以考虑递归解决问题,出口是为空返回另一个连表,递归条件是当当前连表小于另一个连表,返回当前连表并继续判断下一个链表。
func mergeTwoLists(list1 *ListNode, list2 *ListNode) *ListNode {
if list1 == nil {
return list2
}
if list2 == nil {
return list1
}
if list1.Val < list2.Val {
list1.Next = mergeTwoLists(list1.Next, list2)
return list1
} else {
list2.Next = mergeTwoLists(list2.Next, list1)
return list2
}
}