【PAT乙级 1105】链表合并 C++

给定两个单链表 L1​=a1​→a2​→⋯→an−1​→an​ 和 L2​=b1​→b2​→⋯→bm−1​→bm​。如果 n≥2m，你的任务是将比较短的那个链表逆序，然后将之并入比较长的那个链表，得到一个形如 a1​→a2​→bm​→a3​→a4​→bm−1​⋯ 的结果。例如给定两个链表分别为 6→7 和 1→2→3→4→5，你应该输出 1→2→7→3→4→6→5。

输入格式：

输入首先在第一行中给出两个链表 L1​ 和 L2​ 的头结点的地址，以及正整数
N (≤105)，即给定的结点总数。一个结点的地址是一个 5 位数的非负整数，空地址 NULL 用 -1 表示。

随后 N 行，每行按以下格式给出一个结点的信息：

Address Data Next

其中 Address 是结点的地址，Data 是不超过 105 的正整数，Next 是下一个结点的地址。题目保证没有空链表，并且较长的链表至少是较短链表的两倍长。

输出格式：

按顺序输出结果链表，每个结点占一行，格式与输入相同。

输入样例：

00100 01000 7
02233 2 34891
00100 6 00001
34891 3 10086
01000 1 02233
00033 5 -1
10086 4 00033
00001 7 -1

输出样例：

01000 1 02233
02233 2 00001
00001 7 34891
34891 3 10086
10086 4 00100
00100 6 00033
00033 5 -1

代码示例：

#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;

struct node{
    string start;
    int data;
    string next;
};

// //注释掉的这两个函数测试点6一直超时，不知道是因为insert用得太多了还是循环逻辑有问题
// void reverse(vector<node> &link,int len){
//     node temp;
//     for(int p=0;p<len;p++){
//         temp=link[p];
//         link.erase(link.begin()+p);  //注意erase和insert传入迭代器
//         link.insert(link.begin(),temp);
//     }
//     return;
// }
// void combine(vector<node> _long,vector<node> _short,int len_long,int len_short){
//     int p=2,q=0;  //分别为长链表指针和短链表指针
//     while(q<len_short){
//         _long.insert(_long.begin()+p,_short[q]);
//         q++;
//         p+=3;
//     }
//     for(int j=0;j<_long.size()-1;j++)
//         cout<<_long[j].start<<" "<<_long[j].data<<" "<<_long[j+1].start<<endl;
//     cout<<_long[_long.size()-1].start<<" "<<_long[_long.size()-1].data<<" -1";
//     return;
// }


//链表反转
void reverse(vector<node> &link,int len){
    node temp;
    for(int left=0,right=len-1;left<right;left++,right--){
        temp=link[left];
        link[left]=link[right];
        link[right]=temp;
    }
    return;
}

//链表合并
void combine(vector<node> _long,vector<node> _short,int len_long,int len_short){
    vector<node> ans;
    for(int p=0,q=0;p<len_long;p++){
        ans.push_back(_long[p]);
        if(p%2==1 && q<len_short)
            ans.push_back(_short[q++]);
    }
    for(int j=0;j<ans.size()-1;j++)
        cout<<ans[j].start<<" "<<ans[j].data<<" "<<ans[j+1].start<<endl;
    cout<<ans[ans.size()-1].start<<" "<<ans[ans.size()-1].data<<" -1";
    return;
}

int main(){
    string start_l1,start_l2;
    int n;
    cin>>start_l1>>start_l2>>n;
    unordered_map<string,node> nodes;
    node x;
    for(int i=0;i<n;i++){
        cin>>x.start>>x.data>>x.next;
        nodes.insert({x.start,x});
    }

    //找出链表link1和link2
    string i1=start_l1,i2=start_l2;
    vector<node> link1,link2;
    while(nodes.find(i1)!=nodes.end()){
        link1.push_back(nodes[i1]);
        i1=nodes[i1].next;
    }
    while(nodes.find(i2)!=nodes.end()){
        link2.push_back(nodes[i2]);
        i2=nodes[i2].next;
    }

    int len1=link1.size(),len2=link2.size();
    if(len1>=2*len2){
        reverse(link2,len2);
        combine(link1,link2,len1,len2);
    }
    if(len2>=2*len1){
        reverse(link1,len1);
        combine(link2,link1,len2,len1);
    }

    return 0;
}