输入两个递增的链表,单个链表的长度为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},转换过程如下图所示:
思路一 用新链表合并两个链表,循环判断
新建一个链表去合并两个链表,依次插入就好啦。
代码如下:
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* public ListNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pHead1 ListNode类
* @param pHead2 ListNode类
* @return ListNode类
*/
public ListNode Merge (ListNode pHead1, ListNode pHead2) {
// write code here
if (pHead1 == null) {
return pHead2;
}
if (pHead2 == null) {
return pHead1;
}
//这里先新建一个head的前置指针
ListNode head = new ListNode(-1);
ListNode result = head;
//按递增组装新链表
while (pHead1 != null && pHead2 != null) {
if (pHead1.val < pHead2.val) {
result.next = pHead1;
pHead1 = pHead1.next;
} else {
result.next = pHead2;
pHead2 = pHead2.next;
}
result = result.next;
}
//处理链表的结尾
if (pHead1 != null) {
result.next = pHead1;
} else {
result.next = pHead2;
}
//返回head
return head.next;
}
}
复杂度分析:
时间复杂度O(N+M):M N分别表示pHead1, pHead2的长度
空间复杂度O(1):常数级空间
思路二 递归
递归理解起来比较容易
每次判断当前两个链表head的val值,小的留下然后next继续,直到head1或者head2是null进行最后一次赋值就结束啦。
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* public ListNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param pHead1 ListNode类
* @param pHead2 ListNode类
* @return ListNode类
*/
public ListNode Merge (ListNode pHead1, ListNode pHead2) {
// write code here
//新建前置指针
ListNode head = new ListNode(-1);
//递归排序合并
getHead(head,pHead1,pHead2);
return head.next;
}
public void getHead(ListNode pHead, ListNode pHead1, ListNode pHead2) {
if (pHead1 == null) {
pHead.next = pHead2;
return;
}
if (pHead2 == null) {
pHead.next = pHead1;
return;
}
if (pHead1.val < pHead2.val) {
pHead.next = pHead1;
getHead(pHead.next,pHead1.next,pHead2);
}else{
pHead.next = pHead2;
getHead(pHead.next,pHead1,pHead2.next);
}
}
}
看到别人的更加简练的递归代码:
public class Solution {
public ListNode Merge(ListNode list1,ListNode list2) {
// list1 list2为空的情况
if(list1 == null || list2 == null){
return list1 != null ? list1 : list2;
}
// 两个链表元素依次对比
if(list1.val <= list2.val){
// 递归计算 list1.next, list2
list1.next = Merge(list1.next, list2);
return list1;
}else{
// 递归计算 list1, list2.next
list2.next = Merge(list1, list2.next);
return list2;
}
}
}
时间复杂度O(N+M):M N分别表示pHead1, pHead2的长度
空间复杂度O(N+M):迭代次数占用空间