这是链表的第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 := resultfor list1 != nil || list2 != nil {left := math.MaxIntif list1 != nil {left = list1.Val}right := math.MaxIntif 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 := resultfor 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}
}