AtCoder Grand Contest 002 D - Stamp Rally

题意：

一张连通图，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\n",ans[i]);
    return 0;
}