Diamond Heart（dfs）

题目


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;
}