链表合并

75 阅读3分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第16天,点击查看活动详情

题目描述

给定两个单链表 L1=a1→a2→…→an−1→anL1=a1→a2→…→an−1→an 和 L2=b1→b2→…→bm−1→bmL2=b1→b2→…→bm−1→bm,满足:n≥2m。

你的任务是将较短的那个链表逆序,然后将之并入较长的链表,得到形如 a1→a2→bm→a3→a4→bm−1…a1→a2→bm→a3→a4→bm−1… 的结果。

例如给定两个链表分别为 6→76→7 和 1→2→3→4→51→2→3→4→5,你应该输出 1→2→7→3→4→6→51→2→7→3→4→6→5。

补充 本题中可能包含不在两个单链表中的节点,这些节点无需考虑。

输入格式

输入首先在第一行中给出两个链表 L1 和 L2 的头结点的地址,以及正整数 N,即给定的结点总数。

一个结点的地址是一个 5 位数的非负整数(可能包含前导 0),空地址 NULL 用 −1−1 表示。

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

Address Data Next

其中 Address 是结点的地址,Data 是不超过 105 的正整数,Next 是下一个结点的地址。

题目保证没有空链表,并且较长的链表至少是较短链表的两倍长。

输出格式

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

数据范围

1≤N≤105

输入样例:

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

思路

这个题首先要把两个链表分离出来,可以怎么做呢,我们注意到N最大是10的5次方,所以先开两个稍微大于10的5次方的数组,一个数组做当前地址->下一个节点地址的映射,另一个用来存地址对应的值。然后我们通过两个开始地址就可以把两条链表分离出来,分离出来之后反转长度短的,从长度长的从前往后遍历并加入到结果中,每当遍历两个值时,判断另一条链表是否还有值未加入结果链表,如果有,则加入,一直到两条链表所有值都加入到结果中。最后用格式化输出结果即可。

代码

#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_map>
#include <string>using namespace std;
const int N = 1e5 + 10;
int a[N],b[N];
​
​
int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
​
    vector<vector<int>> A,B,res;
    int count = 0;
    int s,e,c,x,y,z;
    cin >> s >> e >> c;
​
    while(c -- )
    {
        cin >> x >> y >> z;
        a[x] = z;
        b[x] = y;
    }
​
    int d = s;
    while(a[d] != -1)
    {
        A.push_back({d,b[d],a[d]});
        d = a[d];
    }
    A.push_back({d,b[d],a[d]});
​
    d = e;
    while(a[d] != -1)
    {
        B.push_back({d,b[d],a[d]});
        d = a[d];
    }
    B.push_back({d,b[d],a[d]});
​
    int n = A.size(),m = B.size(),le = 0;
​
    if(n > m)
    {
        reverse(B.begin(),B.end());
        for(int i = 0;i < n;i ++ )
        {
            res.push_back(A[i]);
            le ++ ;
            if(i % 2 == 1 && count < m)
            {
                B[count][2] = res[le - 1][2];
                res[le - 1][2] = B[count][0];
                res.push_back(B[count ++ ]);
                le ++ ;
            }
        }
    }
    else
    {
        reverse(A.begin(),A.end());
        for(int i = 0;i < m;i ++ )
        {
            res.push_back(B[i]);
            le ++ ;
            if(i % 2 == 1 && count < n)
            {
                A[count][2] = res[le - 1][2];
                res[le - 1][2] = A[count][0];
                res.push_back(A[count ++ ]);
                le ++ ;
            }
​
        }
    }
​
    for(auto x : res)
        if(x[2] != -1) printf("%05d %d %05d\n",x[0],x[1],x[2]);
        else printf("%05d %d %d\n",x[0],x[1],x[2]);
    return 0;
}
​