题意
给出一个带权无向图,每次询问给出边集的一个子集S,问原图中是否存在一棵最小生成树使得S的每条边都在里面。
n,m<=500000,
∑
|S|<=500000
分析
比较裸的lct模板题。
先随便搞出一棵最小生成树,对于每次询问,逐个把边加入。对于新加入的一条边(u,v,w),如果u到v路径上边权的最大值等于w,那么就把该边加入到最小生成树中并删掉最大边,反之则该边不存在于最小生成树中。注意到这里删最大边的话不能删掉询问中的边,那么我们可以每加入一条边后就把该边的权值赋为最小值,最后再恢复过来即可。
后来发现全世界好像就我一个写的lct,的确也跑了差不多2s,可以说是勉强卡过去的。。。
正解是离线+并查集。
考虑一个询问里面某一种权值w的边能否全部被用上。那么我们可以对小于w的边做一次最小生成森林,然后把一个连通块缩成一个点,再看询问里面权值为w的边全部加入后能够构成一个森林。
这样的话我们可以离线来做,把边按照边权分类,然后按边权从小到大来来做,连通性的话用并查集来维护一下即可。
代码(lct)
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<stack>
using namespace std;
const int N=500005;
int n,m,f[N],a[N];
struct data{int x,y,z;}e[N];
struct tree{int l,r,fa,mx,val;bool rev;}t[N*2];
stack<int> sta;
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int find(int x)
{
if (f[x]==x) return x;
else return f[x]=find(f[x]);
}
bool is_root(int x)
{
return x!=t[t[x].fa].l&&x!=t[t[x].fa].r||!t[x].fa;
}
void pushdown(int d)
{
if (!t[d].rev) return;
swap(t[d].l,t[d].r);t[d].rev^=1;
if (t[d].l) t[t[d].l].rev^=1;
if (t[d].r) t[t[d].r].rev^=1;
}
void updata(int x)
{
t[x].mx=max(t[x].val,max(t[t[x].l].mx,t[t[x].r].mx));
}
void remove(int x)
{
while (!is_root(x)) sta.push(x),x=t[x].fa;
sta.push(x);
while (!sta.empty()) x=sta.top(),sta.pop(),pushdown(x);
}
void rttr(int x)
{
int y=t[x].l;
t[x].l=t[y].r;t[t[y].r].fa=x;
if (x==t[t[x].fa].l) t[t[x].fa].l=y;
else if (x==t[t[x].fa].r) t[t[x].fa].r=y;
t[y].fa=t[x].fa;
t[y].r=x;t[x].fa=y;
updata(x);updata(y);
}
void rttl(int x)
{
int y=t[x].r;
t[x].r=t[y].l;t[t[y].l].fa=x;
if (x==t[t[x].fa].l) t[t[x].fa].l=y;
else if (x==t[t[x].fa].r) t[t[x].fa].r=y;
t[y].fa=t[x].fa;
t[y].l=x;t[x].fa=y;
updata(x);updata(y);
}
void splay(int x)
{
remove(x);
while (!is_root(x))
{
int p=t[x].fa,g=t[p].fa;
if (is_root(p))
{
if (x==t[p].l) rttr(p);
else rttl(p);
break;
}
if (x==t[p].l)
if (p==t[g].l) rttr(g),rttr(p);
else rttr(p),rttl(g);
else
if (p==t[g].r) rttl(g),rttl(p);
else rttl(p),rttr(g);
}
updata(x);
}
void access(int x)
{
int y=0;
while (x)
{
splay(x);t[x].r=y;
y=x;x=t[x].fa;
}
}
void make_root(int x)
{
access(x);splay(x);t[x].rev^=1;
}
void link(int x,int y)
{
make_root(y);splay(y);t[y].fa=x;
}
void cut(int x,int y)
{
make_root(x);access(y);splay(y);t[y].l=t[x].fa=0;
}
int query(int x,int y)
{
make_root(x);access(y);splay(y);
while (t[t[y].l].mx==t[y].mx||t[t[y].r].mx==t[y].mx)
if (t[t[y].l].mx>t[t[y].r].mx) y=t[y].l;
else y=t[y].r;
return y;
}
int main()
{
n=read();m=read();
for (int i=1;i<=n;i++) f[i]=i;
t[0].mx=-1;
for (int i=1;i<=m;i++)
{
int x=read(),y=read(),z=read();
e[i].x=x;e[i].y=y;e[i].z=z;t[i+n].val=t[i+n].mx=z;
if (find(x)!=find(y)) f[find(x)]=find(y),link(x,i+n),link(y,i+n);
else
{
int id=query(x,y);
if (t[id].val>z) cut(id,e[id-n].x),cut(id,e[id-n].y),link(x,i+n),link(y,i+n);
}
}
int q=read();
while (q--)
{
int k=read();
for (int i=1;i<=k;i++) a[i]=read();
int flag=0;
for (int i=1;i<=k;i++)
{
int x=e[a[i]].x,y=e[a[i]].y,z=e[a[i]].z;
int id=query(x,y);
if (t[id].val<z) {flag=1;break;}
cut(id,e[id-n].x);cut(id,e[id-n].y);
t[a[i]+n].val=0;
link(a[i]+n,x);link(a[i]+n,y);
}
for (int i=1;i<=k;i++) t[a[i]+n].val=e[a[i]].z,splay(a[i]+n);
if (!flag) puts("YES");
else puts("NO");
}
return 0;
}