题目描述
给定两个单链表 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,即给定的结点总数。
一个结点的地址是一个 5 位数的非负整数(可能包含前导 0),空地址 NULL 用 −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
思路:
首先用数组来模拟链表,可以成立的原因是因为链表的地址都是非负五位数。
然后用vector来整理链表,vector中的结构为pair<int,int>,用来记录该节点的值和地址。
需要比较两个vector的大小,将大的一个定义为ve1,小的为ve2。
将两个vector合并到第三个vector中。
打印出来就可以了。
注意:
有一个细节。博主在这个地方踩坑了。
就是没有用第三个vector而是直接用ve2插入到ve1中(适用insert),这会导致超时!
错误代码部分如下:
p = ve2.size() - 1;
for (int i = 2; i < ve1.size() && p>=0; i += 3)
{
ve1.insert(ve1.begin()+i, ve2[p]);
p--;
}
后来经过改正得到如下代码:
正确代码:
#include<iostream>
#include<vector>
using namespace std;
const int N = 100010;
int v[N], ne[N];
typedef pair<int, int>PII;
int main()
{
ios::sync_with_stdio(false);
int L1, L2;
cin >> L1 >> L2;
int n;
cin >> n;
while (n--)
{
int a, b, c;
cin >> a >> b >> c;
v[a] = b;
ne[a] = c;
}
vector<PII>ve1, ve2,ve3;
int p = L1;
while (p != -1)
{
ve1.push_back({ p,v[p]});
p = ne[p];
}
p = L2;
while (p != -1)
{
ve2.push_back({ p,v[p]});
p = ne[p];
}
if (ve1.size() < ve2.size())
{
swap(ve1, ve2);
}
p = ve2.size() - 1;
for (int i = 0; i < ve1.size(); i++)
{
ve3.push_back(ve1[i]);
if ((i + 1) % 2 == 0)
{
if (p >= 0)
{
ve3.push_back(ve2[p]);
p--;
}
}
}
for (int i = 0; i < ve3.size(); i++)
{
if(i!=0)
printf("%05d\n", ve3[i].first);
printf("%05d %d ", ve3[i].first, ve3[i].second);
}
printf("-1");
}