C/C++合并两个升序链表的方式
合并两个升序链表
算法的思想
1.需要合并的两个链表La,Lb,合并之后的链表Lc(用La的头节点)。
2.定义两个辅助指针Pa,Pb分别是链表La,Lb的复制指针。
3.从首元节点开始比较,当两个链表都没有到达链表尾部的时候,依次取其中较小的数据进行链接到Lc的最后
4.如果两个元素的值相同,取La链的,把Lb链表的元素删除(确保新链表没有重复的元素)
5.当一个链表结束的时候,把非空链表剩余的所有元素链接在Lc表的最后
6.释放Lb的头节点(Lb链表就被删除了)
代码实现+注释
void MergeList(LinkList &La, LinkList &Lb, LinkList &Lc)
{
LinkList pa, pb, pc;
pa = La->next;
pb = Lb->next;
Lc = pc = La;
while (pa && pb)
{
if (pa->data < pb->data)
{ //把小的数据(pa)链接到Lc表上
pc->next = pa;
pc = pa; //保证指向最后的节点上
pa = pa->next; //指针后移
}
else if (pa->data > pb->data)
{ //把小的数据(pb)链接到Lc表上
pc->next = pb;
pc = pb;
pb = pb->next;
}
else
{ //如果两个元素的值相同,取La链的,把Lb链表的元素删
pc->next = pa;
pc = pa;
pa = pa->next;
Lnode *p = pb->next; //保存pb下一个节点
delete (pb);
pb = p;
}
}
pc->next = pa?pa:pb;
delete(Lb);
}
合并K个升序链表(递归方法)
这个题的思路和归并排序的思路一样。
归并的思想
递归地,或迭代地,将两个已经有序的数组或链表合并成一个有序的大数组或大链表。
先来看合并两个有序链表的代码
传入两个链表的头结点,new一个head节点当作合并后的链表的头结点,当两个链表都没有走到链尾的时候,将两链表的节点有序放入合并后的链表中。
当某一个链表已经走到了链尾,此时把另一个链表剩下的部分接到合并后的链表尾部。
返回head节点的下一个节点,因为head节点是new出来的,要记得delete释放掉这个节点的内存。
ListNode* mergeTwo(ListNode* l1,ListNode*l2){
// new一个节点当作合并后的链表的头结点
ListNode* head=new ListNode(0);
ListNode* temp=head;
// 当两个链表都没有走到链尾的时候,将两链表的节点有序放入合并后的链表中
while(l1 && l2){
if(l1->val <= l2->val){
temp->next=l1;
temp=temp->next;
l1=l1->next;
}else{
temp->next=l2;
temp=temp->next;
l2=l2->next;
}
}
// 当某一个链表已经走到了链尾,此时把另一个链表剩下的部分接到合并后的链表尾部。
if(l1){
temp->next=l1;
}else{
temp->next=l2;
}
temp=head->next;
delete head;
head=nullptr;
return temp;
}
我们再来看合并K个链表的递归方法
数组lists内存放了K个链表首节点,我们将数组先分成两部分,再将每部分又分为两部分,直到分成了一个链表首节点为一部分,递归程序就走到底了,然后开始调用合并两个有序链表的函数,将链表两两合并,此时递归程序不断地返回上一级,直到将所有链表合并成一个链表。
class Solution {
public:
// 传入存放了K个链表首节点的数组lists
ListNode* mergeKLists(vector<ListNode*>& lists) {
if(lists.empty()){
return nullptr;
}
return mergeAll(lists,0,lists.size()-1);
}
// 递归地对链表进行两两合并
ListNode* mergeAll(vector<ListNode*>& lists,int begin,int end){
ListNode* l1=nullptr;
ListNode* l2=nullptr;
ListNode* res=nullptr;
if(begin<end){
int mid=(begin+end)/2;
l1=mergeAll(lists,begin,mid);
l2=mergeAll(lists,mid+1,end);
res=mergeTwo(l1,l2);
}else{
return lists[begin];
}
return res;
}
// 合并两个有序链表的函数
ListNode* mergeTwo(ListNode* l1,ListNode*l2){
ListNode* head=new ListNode(0);
ListNode* temp=head;
while(l1 && l2){
if(l1->val <= l2->val){
temp->next=l1;
temp=temp->next;
l1=l1->next;
}else{
temp->next=l2;
temp=temp->next;
l2=l2->next;
}
}
if(l1){
temp->next=l1;
}else{
temp->next=l2;
}
temp=head->next;
delete head;
head=nullptr;
return temp;
}
};
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。
相关文章