蒟蒻做整体二分的第一题。
(好吧其实是第二题,前一次做的那一道我以为是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;
}