[hihocoder]1238 : Total Highway Distance

题目:https://hihocoder.com/problemset/problem/1238?sid=1369284

一道很不错的题目。

思路:对于任意一条边(a,b),权值为w,以a为根的子树节点数目为node[a],以b为根的子树节点node[b],则该边被计算node[a]*node[b]*w次。利用DFS对所有节点进行遍历。

#include 
#include 
#include 
using namespace std;
typedef long long llong;
typedef pair Pair;
const int N=1e6+4;
vector> g(N);
llong total=0LL;
int root=1;
vector level(N,0);
vector node(N,0);
vector path(N,0);
int n,m;
void dfs(int rt){
    node[rt]=1;
    for(auto e:g[rt]){
        if(level[e.first]==0){
            path[e.first]=e.second;
            level[e.first]=level[rt]+1;
            dfs(e.first);
            node[rt]+=node[e.first];
            total+=(llong)node[e.first]*(n-node[e.first])*e.second;
        }
    }
}
void edit(int u,int v,llong w){
    if(level[u]

 

你可能感兴趣的:(算法,hiho刷题日记)