题目描述:
题解:
dfs计数?
#include<cstdio>
#include<string>
using namespace std;
const int maxn=1000005;
int n,tot,lnk[maxn],son[2*maxn],nxt[2*maxn],w[2*maxn];
bool vis[maxn];
long long ans;
inline int read(){
int x=0; char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
return x;
}
void add(int x,int y,int z){
son[++tot]=y,w[tot]=z,nxt[tot]=lnk[x],lnk[x]=tot;
}
int abs_(int x) {if (x<0) return -x; else return x;}
int dfs(int x){
int sum=0;
for (int j=lnk[x];j;j=nxt[j])
if (!vis[son[j]]){
vis[son[j]]=1; int siz=dfs(son[j]); sum+=siz;
ans+=1ll*abs_(n-siz-siz)*w[j];
}
return sum+1;
}
int main(){
n=read();
for (int i=1;i<n;i++) {
int x=read(),y=read(),z=read();
add(x,y,z); add(y,x,z);
}
vis[1]=1; dfs(1);
printf("%lld\n",ans);
return 0;
}