题目:
输入两个递增排序的链表,合并这两个链表并使新链表中的节点仍然是递增排序的。
示例1:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/he-bing-liang-ge-pai-xu-de-lian-biao-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
算法思路
选择两个链表中的某个链表作为基准(main),将另一个链表上的元素按照顺序要求插入到基准链表(exta)中
设定一个基准链表上的当前元素cur,初始化为该表的表头元素。
将另一个链表上的当前比较元素的值和基准链表上的cur的下一个元素作比较
1. 大于等于,则将基准链表(main)当前元素后移,继续比较。
2. 小于,则在另一个链表(extra)上的元素索引往后移动,继续比较,直到条件变成大于等于了,将这段数据插入到main的cur的后面。
比较的终止条件设定为:
1. main上的cur已经是最后一个节点了,将extra上的当前比较元素(以及其后面的所有元素)插入到cur的后面
2. extra上的当前比较元素以及为空。
为了在链表上插入元素,extra链表中需要使用两个索引,一个是p(代表当前等待处理的元素),另一个是本次要插入的最后一个元素(他的下一个元素的值将大于等于cur的值)的索引
代码
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
if (l1 == NULL)
return l2;
else if (l2 == NULL)
return l1;
ListNode* head=l1;
ListNode* cur=l1; //主队列,即开头元素值较小的队列,辅队列往主队列上插入
ListNode *curNext=cur->next;
ListNode* p=l2;//辅队列当前比较节点
if (l1->val > l2->val)
{
head = l2;
cur = head;
curNext = cur->next;
p = l1;
}
while (1)
{
if (curNext == NULL)//主队列已经遍历完,将辅队列剩下元素全部追加上来
{
cur->next = p;
break;
}
if (p == NULL)//辅队列已经遍历完
break;
if (p->val>=curNext->val)//主队列当前节点的值较小,则移动主队列的当前比较节点
{
cur = curNext;
curNext = curNext->next;
}
else
{
ListNode* last = p;
while (last)
{
if (!last->next)
break;
else if (last->next->val >= curNext->val)
break;
last = last->next;
}//找到辅队列中第一个大于等于主队列当前比较值的元素,将其前面的元素全部插入主队列的cur和curnext之间,并移动cur
cur->next = p;
p = last->next;
last->next = curNext;
cur = curNext;
curNext = cur->next;
}
}
return head;
}