Scx117
只一眼,便辽阔了时间。

题意:给你一棵树,求所有点对(i,j)之间链上的最大点权-最小点权的总和?

 

标程:

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 const int N=1000005;
 5 struct node{int u,v,w;}e[N];
 6 int a[N],f[N],sz[N],n;
 7 ll ans;
 8 int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
 9 bool operator < (const node &A,const node &B) {return A.w<B.w;}
10 void solve()
11 {
12   for (int i=1;i<=n;i++) f[i]=i,sz[i]=1;
13   for (int i=1;i<n;i++) e[i].w=max(a[e[i].u],a[e[i].v]);
14   sort(e+1,e+n);
15   for (int i=1;i<n;i++)
16   {
17     int x=find(e[i].u),y=find(e[i].v);
18     if (x==y) continue;
19     ans+=(ll)e[i].w*sz[x]*sz[y];
20     f[x]=y;sz[y]+=sz[x];
21   }
22 }
23 int main()
24 {
25    scanf("%d",&n);
26    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
27    for (int i=1;i<n;i++) scanf("%d%d",&e[i].u,&e[i].v);
28    solve();
29    for (int i=1;i<=n;i++) a[i]=-a[i];
30    solve();
31    printf("%lld\n",ans);
32    return 0;
33 }

 

题解:并查集+技巧

一个思路:按照点权从大到小加点,每次连通块分裂,统计该点相连结的两个连通块的size乘积。

但是分裂很难维护。

所有反过来做就是合并了是吧。并查集。

由于点也很难处理,我们定义边权为该边连接的两个端点的较大权值(以求链上最大为例,最大最小是一样的)。从小到大加边,统计该边两个端点所在的连通块size乘积(这两个连通块的点权都<=该边的权值)。然后合并这两个端点。

posted on 2018-05-28 18:46  Scx117  阅读(139)  评论(0编辑  收藏  举报