程序设计天梯赛 L2-043 龙龙送外卖 dfs树

原创于 2023-04-20 21:05:15 发布 · 399 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法 #图论

程序设计天梯赛专栏收录该内容

10 篇文章

订阅专栏

该代码使用深度优先搜索(dfs)遍历树结构，计算每个节点的深度，并找出深度最大的节点。程序读取树的结构信息，从根节点开始递归遍历，并在过程中更新最大深度。最后，对于每个查询，它计算从给定节点到根节点的路径长度并输出与最大深度的差值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

dfs遍历树，找出树的深度最大的节点。

#include<bits/stdc++.h>
using namespace std;
const int N = 200010;
vector<int> g[N];
int d[N], f[N];
bool st[N];
void dfs(int u)
{
	for(int j=0;j<g[u].size();j++) 
	{
		d[g[u][j]] = d[u] + 1;
		dfs(g[u][j]);
	}
}
int main()
{
	int n, m;
	cin >> n >> m;
	int ro;
	for(int i=1;i<=n;i++)
	{
		cin >> f[i];
		if(f[i] == -1) ro = i;
		else g[f[i]].push_back(i);
	}
	dfs(ro);
	int v = 0;
	int maxn = 0;
	for(int i=1;i<=m;i++)
	{
		int x;
		cin >> x;
		maxn = max(maxn, d[x]);
		while(!st[x]&&x!=ro)
		{
			st[x] = 1;
			v += 2;
			x = f[x];
		}
		cout << v - maxn << endl;
	}
	return 0;
}