PTA Junior（DFS、ベクター）

L2-026ジュニア（25ポイント）

 

分析：

最初は下から上への検索に似ていると思いましたが、これは多くのデータを繰り返し検索してタイムアウトします。

したがって、正しい解決策は上から下への深い検索であり、ベクターコンテナーの導入は非常に重要であり、時間リソースを簡略化するためにスペースリソースが使用されます。

 

コード：

#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;
}