题意:
一张连通图,q次询问从两个点x和y出发,希望经过的点(不重复)数量等于z,经过的边最大编号最小是多少。
整体二分。
并查集维护,如果两个点联通,那么是联通块的size,否则是两个联通块的size和。
#include<bits/stdc++.h>
#define N 200005
using namespace std;
int n,m,q;
int f[N],size[N],aa[N],bb[N];
int find(int x)
{
if(x==f[x])return x;
return find(f[x]);
}
int xx[N],yy[N],zz[N],p[N],ans[N];
int top,st1[N],st2[N],st3[N],ls[N],rs[N];
void solve(int l,int r,int L,int R)
{
if(l==r)
{
for(int i=L;i<=R;i++)ans[p[i]]=l;
int a=find(aa[l]),b=find(bb[l]);
if(size[a]>size[b])swap(a,b);
if(f[a]!=f[b])
{
f[a]=b;
size[b]+=size[a];
}
return ;
}
int mid=(l+r)>>1;
for(int i=l;i<=mid;i++)
{
int a=find(aa[i]),b=find(bb[i]);
if(size[a]>size[b])swap(a,b);
if(f[a]!=f[b])
{
top++;st1[top]=a;st2[top]=b;st3[top]=size[b];
f[a]=b;
size[b]+=size[a];
}
}
int t1=0,t2=0;
for(int i=L;i<=R;i++)
{
int aa=find(xx[p[i]]),bb=find(yy[p[i]]);
int tmp=0;
if(aa!=bb)tmp=size[aa]+size[bb];
else tmp=size[aa];
if(tmp>=zz[p[i]])ls[++t1]=p[i];
else rs[++t2]=p[i];
}
for(int i=L;i<L+t1;i++)p[i]=ls[i-L+1];
for(int i=L+t1;i<=R;i++)p[i]=rs[i-L-t1+1];
while(top)
{
f[st1[top]]=st1[top];
size[st2[top]]=st3[top];
top--;
}
solve(l,mid,L,L+t1-1);
solve(mid+1,r,L+t1,R);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)f[i]=i,size[i]=1;
for(int i=1;i<=m;i++)
{
scanf("%d%d",&aa[i],&bb[i]);
}
scanf("%d",&q);
for(int i=1;i<=q;i++)p[i]=i;
for(int i=1;i<=q;i++)scanf("%d%d%d",&xx[i],&yy[i],&zz[i]);
solve(1,m,1,q);
for(int i=1;i<=q;i++)printf("%d
",ans[i]);
return 0;
}