L2-3 龙龙送外卖---DFS+染色

最新推荐文章于 2023-07-18 08:47:00 发布

_WAWA鱼_

最新推荐文章于 2023-07-18 08:47:00 发布

阅读量625

点赞数 2

分类专栏：比赛补题文章标签：深度优先算法 c++

本文链接：https://blog.csdn.net/qq_51282224/article/details/124502532

版权

比赛补题专栏收录该内容

99 篇文章 6 订阅

订阅专栏

#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
#define int long long
vector<int>g[N];
int father[N],depth[N];
bool col[N]={0};
void dfs(int u)
{
	for(auto v : g[u])
	{
		depth[v]=depth[u]+1;
		dfs(v);
	}
}
signed main()
{
	int n,m;cin>>n>>m;
	int root;
	for(int i=1,x;i<=n;i++)
	{
		cin>>x;
		if(x==-1)root=i;
		else g[x].push_back(i),father[i]=x;
	}
	dfs(root);
	int maxd=0,res=0;
	for(int i=0,x;i<m;i++)
	{
		cin>>x;
		maxd=max(maxd,depth[x]);
		while(x!=root&&!col[x])col[x]=true,x=father[x],res+=2;
		cout<<res-maxd<<'\n';
	}
}