算法提高 分分钟的碎碎念

算法提高 分分钟的碎碎念

以前有个孩子，他分分钟都在碎碎念。不过，他的念头之间是有因果关系的。他会在本子里记录每一个念头，并用箭头画出这个念头的来源于之前的哪一个念头。翻开这个本子，你一定会被互相穿梭的箭头给搅晕，现在他希望你用程序计算出这些念头中最长的一条因果链。

将念头从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;
}