L2-022 重排链表 - 团体程序设计天梯赛-练习集 (pintia.cn)
做了那个L2-002 链表去重后,发现这道题就挺简单了。其实这种链表的题目,也差不多能摸清规律了,本质上不需要用真的链表来做,甚至连STL中的list都不需要,只用一个map,一个vector就可以解决。因为vector用起来比list方便多了,可以说,list能做的,vector都能做。
下次遇见这种需要你存链表的,注意三点:用map构造链表(用于将输出的杂乱节点统合)、用vector存储链表(将逻辑统合后的数据化为物理结构上的统合)、花心思输出链表(你可以直接输出,也可将输出存在其他类型的容器中,最后再输出)
AC代码:
#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;
struct Node
{
int num;
int val;
int next;
};
vector<Node>lst;
int main ()
{
int start, n;
cin >> start >> n;
unordered_map<int, Node>ump;
int a, b, c; //num val next
for (int i = 0; i < n; i++) //先用map将乱序的链表统合
{
cin >> a >> b >> c;
Node temp = { a,b,c };
ump[a] = temp;
}
//将map中的数据放在lst中
int cur = start;
while (cur != -1)
{
lst.push_back(ump[cur]);
cur = ump[cur].next;
}
//输出 同样不需要将next改变,只需要在输出的时候注意一点
int l = 0, r = lst.size() - 1;
int flag = 0; //已经输出的个数
while (l <= r)
{
if (flag %2 == 0)
{
printf("%05d %d %05d\n", lst[r].num, lst[r].val, lst[l].num);
r--;
}
else
{
printf("%05d %d %05d\n", lst[l].num, lst[l].val, lst[r].num);
l++;
}
flag++;
if (flag == lst.size() - 1) //一般情况输出完了
break;
}
//输出最后一个
if (flag % 2 == 0)
printf("%05d %d -1\n", lst[r].num, lst[r].val);
else
printf("%05d %d -1\n", lst[l].num, lst[l].val);
return 0;
}