codeforces1290C Prefix Enlightenment

https://codeforces.com/problemset/problem/1290/C

这题如果改成给出一种方案使得全部点亮，那么就可以2-sat了，因为一个灯泡最多由两个集合控制，我们可以根据灯泡的初始情况判断两者的相互关系，x和y应该同时变换还是不同变换。

然而题意改不得，要问1-i的最优值。2-sat只能求出全部最后的情况，于是看了题解https://www.bilibili.com/video/av86529667?p=5

发现带权并查集可以边合并边求需要按下多少组开关，并查集的权值表示这里面有多少开关是需要按的。

用1-k表示不要按的情况，k+1--2k表示要按的情况。那么初始t[1-k]=0,t[k+1--2k]=1;

按i=1->n从前到后处理，每个连通块内包含的都是到目前位置为止需要在一起的点们。

如果某个位置只由1个集合包含，那么我们可以根据这个位置的灯泡的初始情况判断他必须选择x还是选择x+k，用f[]=0表示另外一种选择不能出现。

那么如果某个位置由2个集合包含，那么如果s[i]=='0'，那么必须要有一个集合选择，如果s[i]=='1'，那么要么全选要么全不选，我们就把他们进行合并。

由于保证有解，所以x+k和x永远不会合并到一个连通块里面，一个联通块里表示你选了一个点，那么这连通块里所有点都要选择。

在求答案的时候，先ans[i]=ans[i-1]，表示此时在1-i-1全部满足的条件下的最优值，然后减去包含这个位置i的两个集合x,y对ans[i-1]的影响calc(x)和calc(y)。calc(x)是选择x还是选择x+k中的最优值。在对x,x+k,y,y+k进行对应的合并后，再加上calc(x)。

#include<bits/stdc++.h>
using namespace std;

const int maxl=3e5+10;

int n,k;
int f[maxl*2],t[maxl*2],ans[maxl];
int a[maxl][2];
char s[maxl];

inline void prework()
{
	scanf("%d%d",&n,&k);
	scanf("%s",s+1);
	int len,x;
	for(int i=1;i<=k;i++)
	{
		scanf("%d",&len);
		for(int j=1;j<=len;j++)
		{
			scanf("%d",&x);
			if(!a[x][0])
				a[x][0]=i;
			else
				a[x][1]=i;
		}
	}
	for(int i=1;i<=2*k;i++)
	{
		f[i]=i;
		if(i>k)
			t[i]=1;
	}
}

inline int find(int x)
{
	if(f[x]!=x)
		f[x]=find(f[x]);
	return f[x];
}

inline void uni(int x,int y)
{
	if(y==0)
		swap(x,y);
	f[y]=x;
	if(x!=0)
		t[x]+=t[y];
}

inline int calc(int x)
{
	int xx=find(x),xk=find(x+k);
	if(xx==0 || xk==0)
		return t[xx+xk];
	return min(t[xx],t[xk]);
}

inline void mainwork()
{
	int x,y,xx,yy,xxk,yyk;
	for(int i=1;i<=n;i++)
	{
		ans[i]=ans[i-1];
		if(!a[i][0]) continue;
		if(!a[i][1])
		{
			x=a[i][0];
			ans[i]-=calc(x);
			if(s[i]=='1')
				f[find(x+k)]=0;
			else
				f[find(x)]=0;
			ans[i]+=calc(x);
		}
		else
		{
			x=a[i][0];y=a[i][1];
			if(s[i]=='1')
			{
				if(find(x)!=find(y))
				{
					ans[i]-=calc(x);
					ans[i]-=calc(y);
					uni(find(x),find(y));
					uni(find(x+k),find(y+k));
					ans[i]+=calc(x);
				}
			}
			else
			{
				if(find(x)!=find(y+k))
				{
					ans[i]-=calc(x);
					ans[i]-=calc(y);
					uni(find(x),find(y+k));
					uni(find(x+k),find(y));
					ans[i]+=calc(x);
				}
			}
		}
	}
}

inline void print()
{
	for(int i=1;i<=n;i++)
		printf("%d\n",ans[i]);
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}