WA求助
查看原帖
WA求助
1038686
Const_AKer楼主2024/8/13 11:25

八倍经验过来的 改了改代码结果WA了

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=100005;
int n,i,u,v,dp[N],siz[N],dep[N],ans=1e18,a;
vector<int> e[N];
void dfs1(int x,int fa)
{
	dep[x]=dep[fa]+1;
	siz[x]=1;
	for(auto &y:e[x])
	{
		if(y!=fa)
		{
			dfs1(y,x);
			siz[x]+=siz[y];
		}
	}
	dp[1]+=dep[x];
}
void dfs2(int x,int fa)
{
	for(auto &y:e[x])
	{
		if(y!=fa)
		{
			dp[y]=dp[x]+siz[1]-siz[y]*2;
			dfs2(y,x);
		}
	}
}
signed main()
{
	cin>>n;
	for(i=1;i<n;i++)
	{
		cin>>u>>v;
		e[u].push_back(v);
		e[v].push_back(u);
	}
	dep[0]=-1;
	dfs1(1,0);
	dfs2(1,0);
	for(i=1;i<=n;i++) if(dp[i]<ans) ans=dp[i],a=i;
	cout<<a;
	return 0;
}
2024/8/13 11:25
加载中...