CF 950C.Zebras 模拟,set

题意  定义一个序列为0开头,0结尾 中间有若干个101010交替的序列为合法. 例如0,010,01010.为合法.

给出01字符串s,将s分成若个个子序列 使得其每个子序列都为合法,并输出任意一种方案或判断无解. |s|<=-1.

//数据:000111000

分别将'0','1'的位置压入集合a,b.

因为0可以单独一个,所以只要把'1'配完即可有解.

每次从'0'开始暴力生成一个子序列,越长越好,边匹配边在set中删除这些数的位置.

若出现0不够 即为无解,后面'1'不够 则重新开始生成序列即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5,mod=1e9+7;
char s[N];
int mk[N];
set<int> a,b;
vector<int> v[N];
int main()
{	
	scanf("%s",s+1);
	int n=strlen(s+1);
	for(int i=1;i<=n;i++)
	{
		if(s[i]=='1')	a.insert(i);
		else	b.insert(i);
	}
	int one=a.size(),num=0;
	bool flag=true;
	while(one)
	{
		int st=0,ls=0;
		while(true)
		{
			if(st%2)
			{
				auto p=a.lower_bound(ls);
				if(p!=a.end())
				{
					int x=(*p);
					a.erase(p),ls=x,one--,v[num].push_back(x);
				}
				else
					break;
			}
			else
			{
				auto p=b.lower_bound(ls);
				if(p!=b.end())
				{
					int x=(*p);
					b.erase(p),ls=(x),v[num].push_back(x),mk[x]=true;
				}
				else
				{
					flag=false;
					break;
				}
			}
			st++;
		}
		if(flag==false)
			break;
		num++;
	}
	if(flag==false)
		puts("-1");
	else
	{
		for(int i=1;i<=n;i++)
			if(!mk[i]&&s[i]!='1')
				v[num].push_back(i),num++;
	
		printf("%d\n",num);
		for(int i=0;i<num;i++)
		{
			printf("%d ",v[i].size());
			for(int j=0;j<v[i].size();j++)
				printf("%d ",v[i][j]);
			printf("\n");
		}
	}
	return 0;
	
}