算法提高 分分钟的碎碎念
以前有个孩子,他分分钟都在碎碎念。不过,他的念头之间是有因果关系的。他会在本子里记录每一个念头,并用箭头画出这个念头的来源于之前的哪一个念头。翻开这个本子,你一定会被互相穿梭的箭头给搅晕,现在他希望你用程序计算出这些念头中最长的一条因果链。
将念头从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
样例说明
最长的因果链有:
1->2->5 (from[5]=2,from[2]=1,from[1]=0)
1->2->7 (from[7]=2,from[2]=1,from[1]=0)
3->4->6 (from[6]=4,from[4]=3,from[3]=0)
3->4->8 (from[8]=4,from[4]=3,from[3]=0)
数据规模和约定1<=n<=1000
思路:
这道题可以用递归来做,定义一个数组arr来存所有的念头,dfs函数有两个参数,n:表示当前的念头,cnt记录长度。从1到N(碎碎念的总长度)循环,若arr[i]=n(第i个念头来自于n),继续递归dsf(i,cnt+1),最后记录cnt即得到最大长度。
代码:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int arr[1001];
int N, mmax = 0;
void dfs(int n, int cnt)
{
int i;
for (i = 1; i <= N; i++)
{
if (arr[i] == n)
dfs(i, cnt + 1);
}
mmax = max(mmax,cnt);
}
int main()
{
cin >> N;
for (int i = 1; i <= N; i++)
{
cin >> arr[i];
}
dfs(0, 0);
cout << mmax << endl;
return 0;
}