PTA 小字辈 （DFS，vector）

L2-026 小字辈 （25 分)

 

分析：

开始以为是类似并查集的从下向上搜索，但是这样会重复搜索很多数据，时间超时。

所以正确解法是从上到下深度搜索，vector容器的引人非常关键，用空间资源简化了时间资源。

 

代码：

#include <bits/stdc++.h>
using namespace std;

int generation[100010], ma;
vector <int> v[100010];

void dfs(int a, int depth)
{
	ma = max(ma, depth);//求出最小代
	generation[a] = depth;
	for (int i = 0; i<v[a].size(); i++)
		dfs(v[a][i], depth + 1);
}

int main()
{
	int n, ancestor;
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		int a;
		cin >> a;
		if (a == -1)ancestor = i;
		else v[a].push_back(i);
	}
	//输入数据，并把每一个人的孩子存进各自的vector

	dfs(ancestor, 1);
	//自祖宗向孩子深搜

	cout << ma << endl;//开始输出
	bool flag = 0;//保证第一个数据前无空格
	for (int i = 1; i <= n; i++){
		if (generation[i] == ma){
			if (flag)cout << ' ' << i;
			else{
				cout << i;
				flag = 1;
			}
		}
	}
	return 0;
}