题目
将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-two-sorted-lists
思路
这个题其实非常简单,本不值一提,但是在代码书写的时候还是有一些技巧需要反复磨练。链表的题增加一个哑节点非常方便,之前这个变量我起名dummy(也是参考别人的),看了官方的起名为preHead,感觉更合适一些,哈哈。
另外,这个题还可以用递归来解,我确实从来没想到过。对于这种简单的题目,能多尝试一个解法其实是更容易去接受的。
代码
迭代法,非常简单,不做解释
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
ListNode dummy = new ListNode();
ListNode index = dummy, index1 = l1, index2 = l2;
while (index1 != null && index2 != null) {
if (index1.val < index2.val) {
index.next = index1;
index1 = index1.next;
} else {
index.next = index2;
index2 = index2.next;
}
index = index.next;
}
index.next = index1 == null ? index2 : index1;
return dummy.next;
}
}
时间复杂度O(m+n),空间复杂度O(1)
递归法其实就两点,(1)递归方向或者说是下一层递归的返回结果是什么(2)递归结束条件。这个题我们可以知道,当我们拿到两个链表的时候,首先可以拿到每个链表的第一个节点,这两个节点较小的那个就是头节点。递归的目的就是去掉这个头节点之后的两个链表合并,返回一个头节点,而返回的这个头节点就是一开始头节点的next节点,这样就连接起来了,也就是找到了递归方向。递归结束条件就是两个链表有一个为空的时候,就不需要递归了,直接返回非空的那个就行。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
if (l1 == null) {
return l2;
} else if (l2 == null) {
return l1;
} else if (l1.val < l2.val) {
l1.next = mergeTwoLists(l1.next, l2);
return l1;
} else {
l2.next = mergeTwoLists(l1, l2.next);
return l2;
}
}
}
时间复杂度O(m+n),因为递归深度最大是两个链表的所有节点数;空间复杂度O(m+n),程序没有声明辅助空间,但是递归需要用到栈,栈的深度最大是m+n。
耗时:26分钟