5分求调,1后的都是re
查看原帖
5分求调,1后的都是re
1112575
with_my_moon楼主2024/6/29 19:22
#include <iostream>
#include <algorithm> 
using namespace std;

int ans=0,t;

struct E{
	int u,v,w;
}e[2000];


struct N{
	int g[2000],cnt=1;
}n[2000];

int dfs(int p,int q){
	int u[p];
	if(n[p].cnt==2) return 1;
	for(int i=1;i<=n[p].cnt-1;i++){
		if(n[p].g[i]==q) continue;
		else{
			u[p]+=dfs(n[p].g[i],p);
		}
	}
	return u[p];
}

int main(){ 
	cin>>t;
	for(int i=1;i<=t-1;i++){
		cin>>e[i].u>>e[i].v>>e[i].w;
		n[e[i].u].g[n[e[i].u].cnt]=e[i].v;
		n[e[i].u].cnt++;
		n[e[i].v].g[n[e[i].v].cnt]=e[i].u;
		n[e[i].v].cnt++;
	}
	for(int i=1;i<=t-1;i++){
		ans+=abs(2*dfs(e[i].v,e[i].u)-t)*e[i].w;
	}
	cout<<ans;
	return 0;
}
2024/6/29 19:22
加载中...