瑞神要考研（山东科技大学第四届ACM校赛）

问题 C: 瑞神要考研

时间限制: 1 Sec 内存限制: 128 MB
提交: 173 解决: 30
[提交][状态][讨论版]

题目描述

瑞神要准备考研了，为了复习数据结构，瑞神在某宝上买了一本数据结构得考研辅导资料《考研数据结构---从入门到放弃》，从此瑞神开始了愉快的复（zhuang）习（bi）。
有一天，瑞神找了好多条链表来辅助自己复习，但是他在复习的过程中一不小心把链表掉在了地上，捡起来的时候链表以及断成了好多个结点，每个结点只保留了当前结点的地址、结点的值和下一个结点的地址。瑞神看着这些结点浑身难受无法复习，为了让瑞神继续复（zhuang）习（bi）下去，请你帮瑞神把链表复原。

输入

第一行给出结点的个数n(1<=n<=100000)。
接下来n行每行给出一个结点的信息：结点地址、值和下一个结点的地址。
地址为5位数字，-1表示地址为NULL。
保证每一条链表的最后一个结点的下一个结点地址为-1。

输出

每一行输出一条链表，只输出每个节点的值。
多条链表按照首结点地址从小到大排序。

样例输入

<span style="background-color:#ccccff">3</span>

<span style="background-color:#ccccff">00322 155 -1</span>

<span style="background-color:#ccccff">00322 87 00323</span>

<span style="background-color:#ccccff">00233 1 -1</span>

样例输出

87 155

提示

校赛感悟： 今天，扎气球扎得很爽。（^.^）

欧拉路径+dfs做一直超时，赛后看大神的代码，幡然醒悟。

思路：用next数组存两顶点先后关系，vis数组标记是否有该顶点信息，in-degree数组存该顶点的入度，然后从1开始循环（要求输出按地址从小到大），寻找入度为0的点，输出val值，然后循环找路径。

附上AC代码：

#include<iostream>
#include<cstring>using namespace std;
const int maxn=1e5+5;
int _next[maxn];
int vis[maxn];
int val[maxn];
int in_degree[maxn];
int n,fr,to,va;int main()
{ios::sync_with_stdio(false);cin>>n;memset(in_degree,0,sizeof(in_degree));memset(_next,0,sizeof(_next));for(int i=0;i<n;i++){cin>>fr>>va>>to;_next[fr]=to;val[fr]=va;vis[fr]=1;if(to!=-1)in_degree[to]++;}for(int i=0;i<maxn;i++){if(vis[i]&&in_degree[i]==0){cout<<val[i];int temp=_next[i];while(temp!=-1){cout<<' '<<val[temp];temp=_next[temp];}cout<<endl;}}return 0;
}

另附上比赛时的TLE代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e5+5;struct edges{
int fr,to,vis,val;
}edge[maxn];int ans[maxn];
int pre,beh,wei;
int top;
int n;
bool cmp(edges a,edges b)
{return a.fr<b.fr;
}void dfs(int u)
{for(int i=0;i<n;i++){if(edge[i].vis==0&&edge[i].fr==u){edge[i].vis=1;if(edge[i].to!=-1){dfs(edge[i].to);ans[top++]=edge[i].val;}else {ans[top++]=edge[i].val;return ;}}}
}
int main()
{scanf("%d",&n);for(int i=0;i<n;i++){scanf("%d%d%d",&edge[i].fr,&edge[i].val,&edge[i].to);edge[i].vis=0;}sort(edge,edge+n,cmp);for(int i=0;i<n;i++){top=0;memset(ans,0,sizeof(ans));if(edge[i].vis==0){dfs(edge[i].fr);for(int j=top-1;j>=0;j--){if(j!=0)printf("%d ",ans[j]);else printf("%d\n",ans[j]);}}}return 0;
}