合并两个排序的链表
描述
输入两个递增的链表,单个链表的长度为n,合并这两个链表并使新链表中的节点仍然是递增排序的。
数据范围:0≤n≤1000,−1000≤节点值≤1000
要求:空间复杂度 O(1),时间复杂度 O(n)
如输入{1,3,5},{2,4,6}时,合并后的链表为{1,2,3,4,5,6},所以对应的输出为{1,2,3,4,5,6}
或输入{-1,2,4},{1,3,4}时,合并后的链表为{-1,1,2,3,4,4},所以对应的输出为{-1,1,2,3,4,4}
示例1
输入:{1,3,5},{2,4,6}
返回值:{1,2,3,4,5,6}
示例2
输入:{},{}
返回值:{}
示例3
输入:{-1,2,4},{1,3,4}
返回值:{-1,1,2,3,4,4}
思路
因为给定了两个有序链表,所以看到这一题的首先可以想到归并排序的思想,维护两个指针,依次选择比较小的值进入链表。又因为这是一个链表相关问题,为了保证无需讨论头节点问题,可以引入一个虚拟头结点dummy。最终的返回值就是dummy.next
需要维护下列变量:
cur:指向当前的结果链表尾,初始为dummy
ptr1:链表1中指针所在位置
ptr2:链表2中指针所在位置
代码
public ListNode Merge(ListNode list1,ListNode list2) {
ListNode ptr1 = list1;
ListNode ptr2 = list2;
ListNode dummy = new ListNode(-1);
ListNode head = null;
ListNode cur = dummy;
while(ptr1!=null && ptr2!=null){
if(ptr1.val<ptr2.val){
cur.next = ptr1;
ptr1=ptr1.next;
}
else{
cur.next = ptr2;
ptr2=ptr2.next;
}
cur = cur.next;
}
if(ptr1!=null){
cur.next=ptr1;
}
if(ptr2!=null){
cur.next=ptr2;
}
return dummy.next;
}
合并k个已排序的链表
描述
合并 k 个升序的链表并将结果作为一个升序的链表返回其头节点。
数据范围:节点总数满足0≤n≤
1
0
5
10^5
105,链表个数满足1≤k≤
1
0
5
10^5
105,每个链表的长度满足1≤len≤200 ,每个节点的值满足∣val∣<=1000
要求:时间复杂度 O(nlogk)
示例1
输入:[{1,2,3},{4,5,6,7}]
返回值:{1,2,3,4,5,6,7}
示例2
输入:[{1,2},{1,4,5},{6}]
返回值:{1,1,2,4,5,6}
思路
同样,遇到该问题的首先思路就是归并排序的思想。
归并排序共需要进行logk次,每次归并的时间复杂度为O(n),因此符合时间复杂度的要求。代码如下:
public class Solution {
public ListNode mergeTwoList(ListNode list1,ListNode list2){//直接复制上面合并两个数组的函数
ListNode ptr1 = list1;
ListNode ptr2 = list2;
ListNode dummy = new ListNode(-1);
ListNode head = null;
ListNode cur = dummy;
while(ptr1!=null && ptr2!=null){
if(ptr1.val<ptr2.val){
cur.next = ptr1;
cur = cur.next;
ptr1=ptr1.next;
}
else{
cur.next = ptr2;
cur = cur.next;
ptr2=ptr2.next;
}
}
if(ptr1!=null){
cur.next=ptr1;
}
if(ptr2!=null){
cur.next=ptr2;
}
return dummy.next;
}
public ListNode mergeKLists(ArrayList<ListNode> lists) {
ArrayList<ListNode> curList=lists;
if(lists.size()==0){//边界情况判断
return null;
}
while(curList.size()>1){
ArrayList<ListNode> newList = new ArrayList<ListNode> ();
for(int i=0;i<curList.size()/2;i++){
newList.add(mergeTwoList(curList.get(2*i),curList.get(2*i+1)));
}
if(curList.size()%2==1){//如果是奇数个的话,需要对最后剩下的一个链表做单独处理
newList.add(curList.get(curList.size()-1));
}
curList = newList;
}
return curList.get(0);
}
}