【国集作业】【AGC002D】Stamp Rally【整体二分】

蒟蒻做整体二分的第一题。
（好吧其实是第二题，前一次做的那一道我以为是CDQ。。。就完美地跳过了这个知识点）
小视野链接
洛谷链接

题目大意

给你一个n个点m条边的无向连通图。有Q组询问，每次询问从两个点x或y出发遍历，使其共覆盖z个点，求得一种方案使得路径上经过的边的最大编号最小。

题解

整体二分（其实没什么好说的。。。）
先离线，每一次搜索吧当前答案小于mid的放左边，否则放右边。可以用启发式并查集判断覆盖的点是否大于z即可。

代码

//#pragma GCC optimize(3)
#include<iostream>
#include<iomanip>
#include<stack>
#include<queue>
#include<list>
#include<vector>
#include<set> 
#include<map>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#define ll long long
#define db double
#define inf 100001
#define INF (int)1e8 
#define mod (int)(1e9+7)
#define pi acos(-1)
#define rd(n) {n=0;char ch;int f=0;do{ch=getchar();if(ch=='-'){f=1;}}while(ch<'0'||ch>'9');while('0'<=ch&&ch<='9'){n=(n<<1)+(n<<3)+ch-48;ch=getchar();}if(f)n=-n;}
using namespace std;
int n,m,qcnt;

struct edge{
    int x,y;
    edge(){}
    edge(int xx,int yy){
        x=xx,y=yy;
    }
}e[inf],st[inf];
int top;

struct question{
    int x,y,cnt,id;
}q[inf],tmpl[inf],tmpr[inf];
int ans[inf];

int fa[inf],siz[inf];

int Root(int k){
    return k==fa[k]?k:Root(fa[k]);
}

void B_Search(int l,int r,int ql,int qr){
    if (l==r){
        for (int i=ql;i<=qr;i++){
            ans[q[i].id]=l;
        }
        int fx=Root(e[l].x),fy=Root(e[l].y);
        if (fx!=fy){
            if (siz[fx]>siz[fy]){
                swap(fx,fy);
            }
            fa[fx]=fy;
            siz[fy]+=siz[fx];
        }
        return;
    }
    int mid=(l+r)/2;
    top=0;
    for (int i=l;i<=mid;i++){
        int fx=Root(e[i].x),fy=Root(e[i].y);
        if (fx!=fy){
            if (siz[fx]>siz[fy]){
                swap(fx,fy);
            }
            fa[fx]=fy;
            siz[fy]+=siz[fx];
            st[++top]=edge(fx,fy);
        }
    }
    int sizl=0,sizr=0;
    for (int i=ql;i<=qr;i++){
        int fx=Root(q[i].x),fy=Root(q[i].y);
        if (fx==fy && siz[fx]>=q[i].cnt){
            tmpl[++sizl]=q[i];
        }
        else if (fx!=fy && siz[fx]+siz[fy]>=q[i].cnt){
            tmpl[++sizl]=q[i];
        }
        else{
            tmpr[++sizr]=q[i];
        }
    }
    for (int i=top;i>=1;i--){
        int fx=st[i].x,fy=st[i].y;
        fa[fx]=fx;
        siz[fy]-=siz[fx];
    }
    for (int i=1;i<=sizl;i++){
        q[ql+i-1]=tmpl[i];
    }
    for (int i=1;i<=sizr;i++){
        q[ql+sizl+i-1]=tmpr[i];
    }
    B_Search(l,mid,ql,ql+sizl-1);
    B_Search(mid+1,r,ql+sizl,qr);
    return;
}

int main(){
    rd(n) rd(m)
    for (int i=1;i<=m;i++){
        rd(e[i].x) rd(e[i].y)
    }
    rd(qcnt)
    for (int i=1;i<=qcnt;i++){
        rd(q[i].x) rd(q[i].y) rd(q[i].cnt)
        q[i].id=i;
    }
    for (int i=1;i<=n;i++){
        fa[i]=i;
        siz[i]=1;
    }
    B_Search(1,m,1,qcnt);
    for (int i=1;i<=qcnt;i++){
        printf("%d\n",ans[i]);
    }
    return 0;
}