树的重心60pts求调

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=105;
struct node
{
	int val;
	vector <int> t;
};
node e[N];
int d[N];
int n;
int cnt=INT_MAX,ans;
int sum;
int dfs(int u,int fa)
{
	int size=1;
	int maxx=0;
	for(auto v:e[u].t)
	{
		if(v==fa) continue;
		int cur=dfs(v,u);
		size+=cur;
		maxx=max(maxx,cur);
	}
	maxx=max(maxx,n-size);
	if(cnt>maxx)
	{
		cnt=maxx;
		ans=u;
	}
	else if(cnt==maxx) ans=min(ans,u);
	return size;
}
void tot(int u,int fa)
{
	for(auto v:e[u].t)
	{
		if(v==fa) continue;
		d[v]=d[u]+1;
		tot(v,u);
	}
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		int w,u,v;
		cin>>w>>u>>v;
		e[i].val=w;
		if(u)
		{
			e[i].t.push_back(u);
			e[u].t.push_back(i);
		}
		if(v)
		{
			e[i].t.push_back(v);
			e[v].t.push_back(i);
		}
	}
	dfs(1,0);
	d[ans]=0;
	tot(ans,0);
	for(int i=1;i<=n;i++)
	{
		sum+=e[i].val*(d[i]);
	}
	cout<<sum;
	return 0;
}