luogu4185 [USACO18JAN]MooTube (并查集)

类似于NOI2018d1t1的离线做法，把询问存下来，排个序，然后倒着给并查集加边，每次询问并查集联通块大小

 1 #include<bits/stdc++.h>
 2 #define ll long long
 3 #define pa pair<int,int>
 4 using namespace std;
 5 const int maxn=100010;
 6 
 7 inline ll rd(){
 8     ll x=0;char c=getchar();
 9     while(c<'0'||c>'9') c=getchar();
10     while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
11     return x;
12 }
13 
14 struct Edge{
15     int a,b,l;
16 }eg[maxn],que[maxn];
17 int N,M;
18 int fa[maxn],siz[maxn],ans[maxn];
19 
20 inline bool cmp(Edge a,Edge b){
21     return a.l>b.l;
22 }
23 
24 int getf(int x){return x==fa[x]?x:fa[x]=getf(fa[x]);}
25 inline void add(int a,int b){
26     int x=getf(a),y=getf(b);
27     fa[x]=y;siz[y]+=siz[x];
28 }
29 
30 int main(){
31     int i,j,k;
32     N=rd(),M=rd();
33     for(i=1;i<N;i++){
34         eg[i].a=rd(),eg[i].b=rd(),eg[i].l=rd();
35     }for(i=1;i<=M;i++){
36         que[i].l=rd(),que[i].a=rd();
37         que[i].b=i;
38     }
39     for(i=1;i<=N;i++) fa[i]=i,siz[i]=1;
40     sort(eg+1,eg+N,cmp);sort(que+1,que+M+1,cmp);
41     for(i=1,j=1;i<=M;i++){
42         for(;j<N&&eg[j].l>=que[i].l;j++) add(eg[j].a,eg[j].b);
43         ans[que[i].b]=siz[getf(que[i].a)];
44     }
45     for(i=1;i<=M;i++) printf("%d\n",ans[i]-1);
46 }

 

转载于:https://www.cnblogs.com/Ressed/p/9661791.html