持续创作,加速成长!这是我参与「掘金日新计划 · 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;
}