P4185 [USACO18JAN] MooTube G 题解

水一篇题解。
也是一道并查集的好题，涉及另一个并查集的基本应用，并查集维护连通块（我跟并查集过不去了？？？）

大致题意：

给你一棵树，对于每次询问求一个点所在连通块中到达该点的最小路径权值大于给定值的点个数。

既然都连通块了，那我们在维护连通块的时候直接不把权值大于 $K$ 的边加进去，用并查集维护，按秩合并，答案就是连通块大小减一。但是考虑对于每次询问都重新操作显然是不行的，所以离线下来按 $K$ 排序。

Code

#include<bits/stdc++.h>
using namespace std;
int n,q,fa[100005],siz[100005],x,y;
struct Edge{
    int u,v,val;
}edge[100005];
struct Quest{
    int k,wh,id,ans;
}quest[100005];
bool cmp(Edge a,Edge b){
    return a.val>b.val;
}
bool cmpq(Quest a,Quest b){
    return a.k>b.k;
}
bool cmqp(Quest a,Quest b){
    return a.id<b.id;
}
int find(int x){
    if(fa[x]==x) return x;
    else return fa[x]=find(fa[x]);
}
int main(){
    scanf("%d %d",&n,&q);
    for(int i=1;i<=n-1;++i) scanf("%d %d %d",&edge[i].u,&edge[i].v,&edge[i].val);
    for(int i=1;i<=q;++i) scanf("%d %d",&quest[i].k,&quest[i].wh),quest[i].id=i;
    sort(edge+1,edge+n,cmp);
    sort(quest+1,quest+q+1,cmpq);
    for(int i=1;i<=n;++i) siz[i]=1,fa[i]=i;
    int j=1;
    for(int i=1;i<=q;++i){
        for(j;edge[j].val>=quest[i].k&&j<n;++j){
        	x=find(edge[j].u),y=find(edge[j].v);
            if(x!=y){
            	fa[x]=y;
            	siz[y]+=siz[x];
			} 
        }
        quest[i].ans=siz[find(quest[i].wh)]-1;
    }
    sort(quest+1,quest+q+1,cmqp);
    for(int i=1;i<=q;++i) printf("%d\n",quest[i].ans);
    return 0;
}