题目链接: http://codeforces.com/contest/915/problem/F
题意:
给你一颗结点带权的树, I ( x , y ) I(x,y) I(x,y) 代表从结点 x x x 到结点 y y y 的路径上最大值减去最小值的差。 要你求 ∑ i = 1 n ∑ j = 1 n I ( i , j ) \sum^{n}_{i=1} \sum^{n}_{j=1} I(i,j) ∑i=1n∑j=1nI(i,j) 。
做法:
最重要的一步是把上面的公式拆分, ∑ i = 1 n ∑ j = 1 n I ( i , j ) = ∑ i = 1 n ∑ j = 1 n m a x ( i , j ) − ∑ i = 1 n ∑ j = 1 n m i n ( i , j ) \sum^{n}_{i=1} \sum^{n}_{j=1} I(i,j)=\sum^{n}_{i=1} \sum^{n}_{j=1} max(i,j)-\sum^{n}_{i=1} \sum^{n}_{j=1} min(i,j) ∑i=1n∑j=1nI(i,j)=∑i=1n∑j=1nmax(i,j)−∑i=1n∑j=1nmin(i,j) 即我们可以把最大值和最小值分开处理。
那么如何求最大值呢(最小值类似不枚举)。不知道有没有做过这样的题目,给你一棵边带权的树,给你 1 e 5 1e5 1e5 个询问,每次询问有多少对结点之间的最大距离不超过 q i q_i qi 的。这样的题目就是从小到大枚举最大值并进行并查集的维护。
这道题的做法也是类似的,如果你要求最大值,那么就从小到大枚举点 i i i ,如果在之前与这个点相连的点 j j j 已经出现过了,那么就表示点 i i i 可以作为端点或者中间值成为集合 j j 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;
}