链表合并 (25分)

链表合并 (25分) PAT2019秋季真题

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

输入格式：
输入首先在第一行中给出两个链表 L ​1 ​​ 和 L ​2 ​​ 的头结点的地址，以及正整数 N (≤10 ​5 ​​ )，即给定的结点总数。一个结点的地址是一个 5 位数的非负整数，空地址 NULL 用 -1 表示。
随后 N 行，每行按以下格式给出一个结点的信息：
Address Data Next 其中 Address 是结点的地址，Data 是不超过 10 ​5 ​​ 的正整数，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

本题注意点：可能会有无效结点

AC代码：

#include<iostream>
#include<map>
#include<algorithm>
#include<vector>
using namespace std;

struct node{
    string first;
    string second;
    int data;
};
vector<node> list1,list2;
map<string,node> list;
int n;
string add1,add2;

int main()
{
    cin >> add1 >> add2 >> n;
    for(int i=0;i<n;i++)
    {
        node p;
        cin >> p.first >> p.data >> p.second;
        list[p.first]=p;
    }
    for(string add=add1;add!="-1";)
    {
        list1.push_back(list[add]);
        add=list[add].second;
    }
    for(string add=add2;add!="-1";)
    {
        list2.push_back(list[add]);
        add=list[add].second;
    }
    vector<node> ans;
    if(list1.size()<list2.size())
    {
        reverse(list1.begin(),list1.end());
        int p=0,q=0;
        for(int i=0;i<list1.size();i++)
        {
            ans.push_back(list2[q++]);
            ans.push_back(list2[q++]);
            ans.push_back(list1[p++]);
        }
        if(list2.size()>2*list1.size())
        {
            while(q<list2.size())
                ans.push_back(list2[q++]);
        }
    }
    else
    {
        reverse(list2.begin(),list2.end());
        int p=0,q=0;
        for(int i=0;i<list2.size();i++)
        {
            ans.push_back(list1[p++]);
            ans.push_back(list1[p++]);
            ans.push_back(list2[q++]);
        }
        if(list1.size()>2*list2.size())
        {
            while(p<list1.size())
                ans.push_back(list1[p++]);
        }
    }
    for(int i=0;i<ans.size()-1;i++)
        printf("%s %d %s\n",ans[i].first.c_str(),ans[i].data,ans[i+1].first.c_str());
    printf("%s %d -1",ans[ans.size()-1].first.c_str(),ans[ans.size()-1].data);
    return 0;
}