题目
Sample1
输入
3
1 2 1
2 3 2
输出
6
Sample2
输入
10
1 2 3
1 3 4
1 4 5
2 5 8
3 6 7
6 7 10
7 8 12
8 9 13
3 10 2
输出
966
题解
注意到,此题如果去枚举dist,那么复杂度最优也是O(n^2logn)的。
我们考虑dist的意义。对于每一条边,这条边两端上,左半边的点与右半边的点,一定会经过这条边。所以这条边会被经过左半边点数*右半边点数,这条边的权值是w,所以这条边产生的答案贡献就是次数乘权值。
把所有边的贡献加起来就是答案了。
Code
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define IO ios::sync_with_stdio(false);cin.tie(0)
map<pair<int,int>,int>W;
vector<int>G[100005];
bool vis[100005];
int n;ll ans;
ll dfs(int k)//每次返回k点的右半边点数
{
vis[k]=1;
ll sum=0;
for(int i=0;i<G[k].size();i++){
if(vis[G[k][i]])continue;
ll t=dfs(G[k][i]);
ans+=(n-t)*t*W[{k,G[k][i]}]; //ans+=左半边点数*右半边点数*当前边的边权
sum+=t;
}
return sum+1;//+1为结点k本身
}
int main()
{
IO;
int u,v,w;
cin>>n;
for(int i=1;i<n;i++){
cin>>u>>v>>w;
W[{u,v}]=W[{v,u}]=w;
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1);
cout<<ans<<endl;
return 0;
}