题目大意说明:
一个无向图,每条边的代价是 边长度*边两侧的点数的绝对值。求整个图的代价
解题说明:问题在于怎么求出边两端的点数~
AC代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=1e6+10;
ll ans=0;
struct edge{
int v,next,len;
}E[MAXN<<2];
int p[MAXN],eid,n;
void init(){
memset(p,-1,sizeof(p));
eid=0;
}
void inst(int u,int v,int w){
E[eid].v=v;
E[eid].len=w;
E[eid].next=p[u];
p[u]=eid++;
}
ll dfs(int star,int last){
ll re=1;
for(int i=p[star];i+1;i=E[i].next){
int v=E[i].v;
if(v==last)continue;
ll temp=dfs(v,star);
re+=temp;
ans+=(ll)E[i].len*(ll)abs(n-2*temp);
}
return re;
}
int main(){
init();
scanf("%d",&n);
for(int i=1;i<n;i++){
int ip1,ip2,ip3;
scanf("%d%d%d",&ip1,&ip2,&ip3);
inst(ip1,ip2,ip3);
inst(ip2,ip1,ip3);
}
dfs(1,0);
printf("%lld\n",ans);
return 0;
}