著名的王牌间谍 007 需要执行一次任务,获取敌方的机密情报。已知情报藏在一个地下迷宫里,迷宫只有一个入口,里面有很多条通路,每条路通向一扇门。每一扇门背后或者是一个房间,或者又有很多条路,同样是每条路通向一扇门…… 他的手里有一张表格,是其他间谍帮他收集到的情报,他们记下了每扇门的编号,以及这扇门背后的每一条通路所到达的门的编号。007 发现不存在两条路通向同一扇门。
内线告诉他,情报就藏在迷宫的最深处。但是这个迷宫太大了,他需要你的帮助 —— 请编程帮他找出距离入口最远的那扇门。
输入格式:
输入首先在一行中给出正整数 N(<105),是门的数量。最后 N 行,第 i 行(1≤i≤N)按以下格式描述编号为 i 的那扇门背后能通向的门:
K D[1] D[2] ... D[K]
其中 K
是通道的数量,其后是每扇门的编号。
输出格式:
在一行中输出距离入口最远的那扇门的编号。题目保证这样的结果是唯一的。
输入样例:
13
3 2 3 4
2 5 6
1 7
1 8
1 9
0
2 11 10
1 13
0
0
1 12
0
0
输出样例:
12
思路:首先要找到入口,很显然,这里每条边实际上是有向的,每扇门指向其后面的门,所以入口就是没有被任何一扇门指向的那扇门(即:n行中没有出现的编号,即为入口编号) ,然后dfs找到距离最远的点即可(dfs时记录深度len,更新最大深度maxlen的同时更新距离最远的点的编号point即可),由于这个题实际上是一个不存在平行边的有向根树,入口即为树根,所以dfs时可以不用vis数组标记遍历过的点(dfs中vis数组标记多数用于避免访问重边(而这里没有重边,所以不用)),一开始用的vis数组+depth数组(记录每个点深度),后来发现这样可以做但完全没必要,只需要在dfs中加一个参数记录当前深度即可。
初始AC代码:
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+5;
vector<int>v[maxn];
int n;
int maxlen=-1,point,vis[maxn],depth[maxn];
void dfs(int x)
{
vis[x]=1;
for(int i=0;i<v[x].size();i++){
int p=v[x][i];
if(!vis[p]){
vis[p]=1;
depth[p]=depth[x]+1;
if(depth[p]>maxlen){
maxlen=depth[p];
point=p;
}
dfs(p);
vis[p]=0;
}
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n;
for(int i=1;i<=n;i++){
int k;
cin>>k;
while(k--){
int x;
cin>>x;
v[i].push_back(x);
vis[x]=1;
}
}
if(n==1){
cout<<1<<endl;
return 0;
}
int start;
for(int i=1;i<=n;i++){
if(!vis[i]){
start=i;
break;
}
}
memset(vis,0,sizeof(vis));
dfs(start);
cout<<point<<endl;
return 0;
}
简化后的AC代码:
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+5;
vector<int>v[maxn];
int n,vis[maxn];
int maxlen=-1,point;
void dfs(int x,int len)
{
len++;
if(len>maxlen){
maxlen=len;
point=x;
}
for(int i=0;i<v[x].size();i++){
dfs(v[x][i],len);
}
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n;
for(int i=1;i<=n;i++){
int k;
cin>>k;
while(k--){
int x;
cin>>x;
v[i].push_back(x);
vis[x]=1;
}
}
int start;
for(int i=1;i<=n;i++){
if(!vis[i]){
start=i;
break;
}
}
dfs(start,0);
cout<<point<<endl;
return 0;
}