AcWing 4273. 链表合并

问题描述

给定两个单链表$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，即给定的结点总数。
一个结点的地址是一个 5 位数的非负整数（可能包含前导 0），空地址 NULL 用 −1 表示。
随后 N 行，每行按以下格式给出一个结点的信息：Address Data Next
其中 Address 是结点的地址，Data 是不超过$10^5$的正整数，Next 是下一个结点的地址。
题目保证没有空链表，并且较长的链表至少是较短链表的两倍长。

输出格式

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

数据范围

1≤N≤$10^5$

输入样例：

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

解题思路

第一种解法

因为最后要输出合并后的链表，所以要保存每个结点的地址和值，就会想到用pair，输入数据后，将两个链表各自连接起来，分别为a和b，长的链表为a，短的为b（注意顺序）然后长的链表插入两个结点，短的链表插入一个结点，这样依次插入，还要特判结点够不够。

AC代码

#include <iostream>
#include <vector>
using namespace std;
#define x first
#define y second
typedef pair<int,int> PII;
const int N=100010;
int v[N],ne[N];
int main()
{
    int h1,h2,n;
    cin>>h1>>h2>>n;
    while(n--)
    {
        int addr,data,next;
        cin>>addr>>data>>next;
        v[addr]=data;
        ne[addr]=next;
    }
    vector<PII> a,b;
    for(int i=h1;i!=-1;i=ne[i])
    a.push_back({i,v[i]});
    for(int i=h2;i!=-1;i=ne[i])
    b.push_back({i,v[i]});
    if(a.size()<b.size()) swap(a,b);//令a为长链表
    vector<PII> c;
    for(int i=0,j=b.size()-1;i<a.size();i+=2,j--)
    {
        c.push_back(a[i]);
        if(i+1<a.size()) c.push_back(a[i+1]);//第二个数存在时加入c中
        if(j>=0) c.push_back(b[j]);
    }
    for(int i=0;i<c.size();i++)
    {
        printf("%05d %d ",c[i].x,c[i].y);
        if(i+1<c.size()) printf("%05d\n",c[i+1].x);
        else puts("-1");
    }
    return 0;
}

第二种解法

用v[addr]表示地址为addr的Data,ne[addr]存的是下一个结点的地址。然后用两个vector:a和b存两条链表（a为长的链表）。
最后用一个vector:ans存结果链表。

AC代码

#include <iostream>
#include <vector>
using namespace std;
const int N=100010;
int v[N],ne[N];
int main()
{
    int h1,h2,n;
    cin>>h1>>h2>>n;
    for(int i=0;i<n;i++)
    {
        int addr;
        cin>>addr;
        cin>>v[addr]>>ne[addr];
    }
    vector<int> a,b;
    for(int i=h1;i!=-1;i=ne[i])
    a.push_back(i);
    for(int i=h2;i!=-1;i=ne[i])
    b.push_back(i);
    vector<int> ans;
    if(a.size()<b.size()) swap(a,b);
    for(int i=0,j=b.size()-1;i<a.size();i+=2,j--)
    {
        ans.push_back(a[i]);
        if(i+1<a.size()) ans.push_back(a[i+1]);
        if(j>=0) ans.push_back(b[j]);
    }
    for(int i=0;i<ans.size();i++)
    {
        printf("%05d %d ",ans[i],v[ans[i]]);
        if(i+1<ans.size()) printf("%05d\n",ans[i+1]);
        else puts("-1");
    }
    return 0;
}