题意:
给出一个有n个点的带权值的树,求所有点两两之间的所有路径所经过的最大值与最小值的差。
算法:
并查集
解题思路:
没怎么做过这种类型的题,首先考虑每一个点对答案做出的贡献,可以求出所有最大值的和以及所有最小值的和。这里介绍一下怎么求所有最大值的和,首先我们有一个完整的树,首先考虑权值最大的那个点,肯定是和他相连的并且两两相互独立的几坨点的个数两两相乘后乘以这个点权(这里有点求解的技巧,详细看代码)。然后将这个点去掉,并且删除所有和他相邻的边,继续考虑第二大的点。然而可以发现复杂度太高,行不通。那么便逆向思维,首先将所有点标记成不能用,然后从权值最小的开始标记成能用,其余的方法不变。
最小值求解和最大值一样,不再赘述。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#define LL long long
using namespace std;
const int Max = 2e6+100;//T了一次是因为数组开太小。
int tree[Max], head[Max], nxt[Max], point[Max], tot;
bool vis[Max];
struct node{
int id, val;
bool operator <(const node &s)const{
return val<s.val;
}
}data[Max];
void add_edge(int u, int v){
point[tot]=v, nxt[tot]=head[u], head[u]=tot++;
}
int get_root(int root){
return tree[root]==root?root:tree[root]=get_root(tree[root]);
}
LL ans;
int size_tree[Max];//以某个点为根的有多少个点
void work(int u, int val){
for(int i=head[u]; i!=-1; i=nxt[i]){
int v=point[i];
if(!vis[v]){
int x=u, y=get_root(v);
ans+=1ll*size_tree[x]*size_tree[y]*val;//那个所谓的技巧
size_tree[x]+=size_tree[y], size_tree[y]=1;//并查集过程随便维护点数
tree[y]=x;
}
}
}
void solve(int n){
for(int a=1; a<=n; a++)tree[a]=a, size_tree[a]=1;
sort(data, data+n);
for(int a=1; a<=n; a++)vis[a]=true;
for(int a=0; a<n; a++){
vis[data[a].id]=false;
work(data[a].id, data[a].val);
}
}
int main(){
int n, u, v;
scanf("%d", &n);
memset(head, -1, sizeof(head));
for(int a=0; a<n; a++)scanf("%d", &data[a].val), data[a].id=a+1;
for(int a=0; a<n-1; a++)scanf("%d%d", &u, &v), add_edge(u, v), add_edge(v, u);
solve(n);
for(int a=0; a<n; a++)data[a].val=-data[a].val;//所谓的技巧二
solve(n);
printf("%lld\n", ans);
return 0;
}