题目描述
以前有个孩子,他分分钟都在碎碎念。不过,他的念头之间是有因果关系的。他会在本子里记录每一个念头,并用箭头画出这个念头的来源于之前的哪一个念头。翻开这个本子,你一定会被互相穿梭的箭头给搅晕,现在他希望你用程序计算出这些念头中最长的一条因果链。
将念头从1到n编号,念头i来源于念头from[i],保证from[i]<i,from[i]=0表示该念头没有来源念头,只是脑袋一抽,灵光一现。
输入格式
第一行一个正整数n表示念头的数量
接下来n行依次给出from[1],from[2],…,from[n]
输出格式
共一行,一个正整数L表示最长的念头因果链中的念头数量
样例输入
8
0
1
0
3
2
4
2
4
样例输出
3
分析
其实就是要求我们构建一棵树,然后求这棵树的高度。
C++ 代码
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
string s;
int n,x,ans;
vector<int> v[1010];
void dfs(int u,int layer) //求树的高度
{
if(v[u].size()<1)
{
ans=max(ans,layer);
return ;
}
for(auto x:v[u])
{
dfs(x,layer+1);
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>x;
v[x].push_back(i);
}
for(int i=0;i<=n;i++)
{
dfs(i,0);
}
cout<<ans;
return 0;
}