PTA Junior (DFS, vector)

L2-026 Junior (25 puntos)

 

análisis:

Al principio pensé que era similar a buscar de abajo hacia arriba, pero esto repetidamente buscaría una gran cantidad de datos y se agotaría el tiempo.

Por tanto, la solución correcta es una búsqueda profunda de arriba a abajo, la introducción de contenedores vectoriales es muy importante y los recursos espaciales se utilizan para simplificar los recursos de tiempo.

 

Código:

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