915 F. Imbalance Value of a Tree 并查集

题目链接： http://codeforces.com/contest/915/problem/F

题意：

给你一颗结点带权的树， $I(x,y)$ 代表从结点 $x$ 到结点 $y$ 的路径上最大值减去最小值的差。 要你求 ${\sum }_{i=1}^n{\sum }_{j=1}^{n}I(i,j)$ 。

做法：

最重要的一步是把上面的公式拆分，${\sum }_{i=1}^n{\sum }_{j=1}^{n}I(i,j)={\sum }_{i=1}^n{\sum }_{j=1}^{n}max(i,j)-{\sum }_{i=1}^n{\sum }_{j=1}^{n}min(i,j)$ 即我们可以把最大值和最小值分开处理。

那么如何求最大值呢（最小值类似不枚举）。不知道有没有做过这样的题目，给你一棵边带权的树，给你 $1e5$ 个询问，每次询问有多少对结点之间的最大距离不超过 $q_i$ 的。这样的题目就是从小到大枚举最大值并进行并查集的维护。

这道题的做法也是类似的，如果你要求最大值，那么就从小到大枚举点 $i$ ，如果在之前与这个点相连的点 $j$ 已经出现过了，那么就表示点 $i$ 可以作为端点或者中间值成为集合 $j$ 的最大值的一部分，（如果还有更大的值那么当前一定还没被枚举到，所以不会被计算）。 那么就直接这样依次做下来就好了。

最小值同理从大做到小即可。

代码

#include <bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep_e(i,u) for(int i=head[u];~i;i=nex[i])
using namespace std;
typedef long long ll;
const int maxn=1000005;
const int maxm=2000005;
int n,fa[maxn],vis[maxn],sz[maxn];
int head[maxn],to[maxm],nex[maxm],cnt;
struct node{
    int val,id;
}e[maxn];
bool cmp(node a,node b){
    return a.val<b.val;
}
int fin(int x){
    return fa[x]==x?x:fa[x]=fin(fa[x]);
}
void add(int u,int v){
    to[cnt]=v;nex[cnt]=head[u];
    head[u]=cnt++;
}
void init(){
    rep(i,1,n) fa[i]=i,vis[i]=0,sz[i]=1;
}
ll deal(int st,int len,int en){
    init();
    ll ret=0;
    for(int j=st;j!=en;j+=len){
        int u=e[j].id;
        ll sum=0;
        rep_e(i,u){
            int v=to[i];
            if(vis[v]){
                v=fin(v);
                sum+=sz[v];
            }
        }
        sum++;
        rep_e(i,u){
            int v=to[i];
            if(vis[v]){
                v=fin(v);
                ret+=(ll)(sum-sz[v])*sz[v]*e[j].val;
                fa[v]=u; sz[u]+=sz[v];
                sum-=sz[v];
            }
        }
        vis[u]=1;
    }
    return ret;
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    rep(i,1,n) scanf("%d",&e[i].val),e[i].id=i;
    rep(i,2,n) {
        int x,y; scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    sort(e+1,e+1+n,cmp);
    ll maxs=deal(1,1,n+1);
    ll mins=deal(n,-1,0);
    printf("%lld\n",maxs-mins);
    return 0;
}