BM4 合并两个排序的链表
输入两个递增的链表,单个链表的长度为n,合并这两个链表并使新链表中的节点仍然是递增排序的。
数据范围: 0 \le n \le 10000≤n≤1000,-1000 \le 节点值 \le 1000−1000≤节点值≤1000
要求:空间复杂度 O(1)O(1),时间复杂度 O(n)O(n)
如输入{1,3,5},{2,4,6}时,合并后的链表为{1,2,3,4,5,6},所以对应的输出为{1,2,3,4,5,6},转换过程如下图所示:
输入:
{1,3,5},{2,4,6}
返回值:
{1,2,3,4,5,6}
思路很简单,就是新建一个节点,由这个节点往后添加节点即可,期间不断比较两个链表所指节点的大小,看应该指向哪个节点。
需要注意的是,一开始就进行判断,如果p1或者p2为null,那么就直接可以将p指向另一个不为null的链表的头节点。
如果两者均为null,当然ans初始值也是null,并且不会被更新,会直接返回null。
先将头节点判断出来并保存,方便返回。
然后就是比较节点值的大小;
最后将没有遍历到的节点统一挪过来,直接就让p.next指向没移动完的链表的当前节点就ok。
做题过程中,弄混了p.next=p1和p=p.next,两者是不一样的,前者是将p的后一个节点赋值,后者是p向后走一个节点。
/**
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
/**
*
* @param pHead1 ListNode类
* @param pHead2 ListNode类
* @return ListNode类
*/
struct ListNode* Merge(struct ListNode* pHead1, struct ListNode* pHead2 ) {
// write code here
if(pHead1==NULL)
return pHead2;
else if(pHead2==NULL)
return pHead1;
struct ListNode* p=NULL;
struct ListNode* ans=NULL;
if (pHead1->val <= pHead2->val) {
p = pHead1;
pHead1 = pHead1->next;
} else {
p = pHead2;
pHead2 = pHead2->next;
}
ans=p;
while (pHead1 != NULL && pHead2!= NULL) {
if (pHead1->val <= pHead2->val) {
p->next = pHead1;
pHead1 = pHead1->next;
p = p->next;
}
else {
p->next = pHead2;
pHead2 = pHead2->next;
p = p->next;
}
}
if(pHead1==NULL||pHead2==NULL)
{
if(pHead1!=NULL)
{
p->next=pHead1;
}
if(pHead2!=NULL)
{
p->next=pHead2;
}
}
return ans;
}