写在前面
写这个是之前听说有人猿辅导面试让手撕这道题,不能用循环,这就逼着大家用递归来做。如果换做是我的话肯定也十分困难,因为我之前做的是用迭代做的,没有仔细看递归,劝大家有多法的题一定要多看看,尽量掌握。(庆幸没遇到这个题)今日记录一下。
合并两个有序链表
先别一口吃个大胖子,先从简单的开始,合并两个排序好的链表。来自LeetCode21
题目描述:将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4
题解及思路
迭代应该很好想,先新建一个哨兵,依次判断两个节点的值的大小,小的连接到哨兵后边,指针后移,不断循环。
public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
ListNode pre = new ListNode(-1);
ListNode cur = pre;
while (l1 != null && l2 != null) {
if (l1.val < l2.val) {
cur.next = l1;
l1 = l1.next;
} else {
cur.next = l2;
l2 = l2.next;
}
cur = cur.next;
}
if (l1 == null) {
cur.next = l2;
} else {
cur.next = l1;
}
return pre.next;
递归写法,一般写递归都要先考虑递归终止条件,容易发现,当一个节点为null时那么就可以直接返回另一个节点即可。后边的逻辑就是判断两个节点的值,修改小的节点指针指向,递归返回。代码如下:
/**
* 递归合并两个排序链表
*
* @param list1
* @param list2
* @return
*/
private ListNode merge2ListNode(ListNode list1, ListNode list2) {
if (list1 == null) return list2;
if (list2 == null) return list1;
if (list1.val < list2.val) {
list1.next = merge2ListNode(list1.next, list2);
return list1;
} else {
list2.next = merge2ListNode(list1, list2.next);
return list2;
}
}
合并K个有序链表
有了合并两个排序链表的思路,那么合并k个的话自然可以想到迭代两个合成一个,然后再往后边合并。这里也可用递归的办法进行合并。类似归并排序的思想,不断切分,然后借助于合并两个链表的函数进行归并。
public ListNode mergeKLists(ListNode[] lists) {
if (lists == null || lists.length == 0) return null;
return merge(lists, 0, lists.length - 1);
}
private ListNode merge(ListNode[] lists, int low, int high) {
if (low == high) return lists[low];
int mid = (low + high) >> 1;
ListNode l1 = merge(lists, low, mid);
ListNode l2 = merge(lists, mid + 1, high);
return merge2ListNode(l1, l2);
}
最后还是想说一句,方法能尽量多掌握就多掌握。