并查集总结(题目合集)

1、codeforces 1213G Path Queries

题意:给你一颗n个点的树,求路径最大边权小于等于某个数(m组)的点对数,n,m<=2e5。

分析:算是一道思维题,想到了联通块计算贡献,不知道怎么处理多组询问,看了题解才知道是先将边权排序,依次加入加入边,不断合并并计算联通块的贡献,注意路径压缩,否则tle。

代码:

#include
using namespace std;
#define ll long long
const int N = 1e6+5;
int n,m,wt[N],f[N],sz[N];
struct nd{
    int u,v,w;
}p[N];
bool cmp(nd x,nd y) {
    return x.w pr[N];

int fd(int x) {
    if(x==f[x]) return x;
    return f[x] = fd(f[x]);
}
ll ans[N];
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++) f[i]=i,sz[i]=1;
    for(int i=1;i>p[i].u>>p[i].v>>p[i].w;
    }
    sort(p+1,p+n,cmp);
    for(int i=1;i<=m;i++) {
        cin>>wt[i];
        pr[i] = {wt[i],i};
    }
    ll res = 0;
    sort(pr+1,pr+1+m);
    int i=1;
    for(int j=1;i<=m&&j

 

你可能感兴趣的:(并查集总结(题目合集))