本地过了洛谷 RE
查看原帖
本地过了洛谷 RE
601764
kele7楼主2024/6/1 16:00

记录

下载了测试点 1,但是本地正常输出,请问这是怎么会是?

#include<bits/stdc++.h>
using namespace std;
int n,uu,vv,dp[1000005],siz[1000005],maxn,place;
vector<int>g[1000005];
int dfs1(int u,int fa,int deep){
	dp[1]+=deep;
	siz[u]=1;
	for(int i=0;i<g[u].size();i++){
		if(g[u][i]==fa)continue;
		dfs1(g[u][i],u,deep+1);
		siz[u]+=siz[g[u][i]];
	}
}
int dfs2(int u,int fa){
	for(int i=0;i<g[u].size();i++){
		if(g[u][i]==fa)continue;
		dp[g[u][i]]=dp[u]+n-2*siz[g[u][i]];
		dfs2(g[u][i],u);
	}
}
int main(){
	cin>>n;
	for(int i=1;i<n;i++){
		cin>>uu>>vv;
		g[uu].push_back(vv);
		g[vv].push_back(uu);
	}
	dfs1(1,0,0);
	dfs2(1,0);
	for(int i=1;i<=n;i++){
		if(dp[i]>maxn){
			maxn=dp[i];
			place=i;
		}
	}
	cout<<place;
}
2024/6/1 16:00
加载中...