abc214d Sum of Maximum Weights

题面

题意

• 给定一个 $n$ 点的树，边有权值，定义 $f(u,v)$ 是 $u$ 到 $v$ 最短路上的最大权边。求：
  $$\sum _{i=1}^{n-1}\sum _{j=i+1}^{n}f(i,j)$$

• $2\le n\le 10^5,1\le w_i\le 10^7$

思路

1. 正难则反，考虑每条边的贡献的时候，左右两边点数如果是删边很难维护，那么考虑加边操作。从小到大加边，类似最小生成树过程，每次加边会连通两个集合，这两个集合所有边都小于该边，于是加上该边的贡献：两个集合点数乘积乘以边权

代码

const int N = 1e5 + 5;

int n;
int fa[N], siz[N];

void init() {for(int i = 0; i < N; ++i) fa[i] = i, siz[i] = 1;}

int find(int x) {return x == fa[x] ? x : fa[x] = find(fa[x]);}

void uni(int x, int y) {
    x = find(x), y = find(y);
    if(x != y) fa[x] = y, siz[y] += siz[x];
}

struct edge {
    int u, v, w;
    friend bool operator < (edge a, edge b) {return a.w < b.w;}
};

edge e[N];

void inp() {
    cin >> n;
    rep(i, 1, n - 1) e[i] = {rd, rd, rd};
    sort(e + 1, e + n);
}

int main() {
    ll ans = 0;
    inp();
    init();
    rep(i, 1, n - 1) {
        int u = find(e[i].u), v = find(e[i].v);
        ans += (ll)siz[u] * siz[v] * e[i].w;
        uni(u, v);
    }
    cout << ans << endl;
    return 0;
}