CF949A Zebras

思路

我们可以先模拟一下：

如果发现该位为 \(1\)，那么只能将这一位放在目前末位是 \(0\) 的子序列。如果不存在，则无解；

如果发现该为为 \(0\)，那么要么将这一位放在目前末位是 \(1\) 的子序列，要么新开一个子序列，并放在首位。

如果最后存在子序列的末位是 \(1\) 在该种分法不合理，因为没有要求子序列的个数，所以可以采取贪心策略，尽可能的减少末位 \(1\) 的子序列个数，如果当前为 \(0\) 且存在末位为 \(1\) 的子序列，则放在该子序列的末位，不存在，才新开一个子序列。

我们可以用一个变量 \(cnt\) 储存目前的子序列个数，用一个变量 \(num\) 储存目前末位为 \(1\) 的子序列个数。

每次遇到 \(1\) 就把 \(1\) 放在第 \(num+1\) 个子序列的末位，如果 \(num=cnt\) 则代表非法。

每次遇到 \(0\) 则把 \(0\) 放在第 \(num\) 个子序列的末位，如果 \(num=0\) 则代表需要新开一个子序列。

最后判断 \(num\) 是否为 \(0\) 就可以判断能否划分了。

然后思考如何输出方案，可以使用 vector 储存每个子序列的情况。

AC code

#include<bits/stdc++.h>
using namespace std;
int cnt,num,n;
vector<int>v[200005];
char s[200005];
int main()
{
	scanf("%s",s),n=strlen(s);
	for(int i=0;i<n;++i)
	{
		if(s[i]=='1')
		{
			if(num==cnt) printf("-1"),exit(0);
			v[++num].push_back(i);
		}
		else
		{
			if(num) v[num--].push_back(i);
			else v[++cnt].push_back(i);
		}
	}
	if(!cnt||num) printf("-1"),exit(0);
	printf("%d\n",cnt);
	for(int i=1;i<=cnt;++i)
	{
		printf("%d",v[i].size());
		for(int j=0;j<v[i].size();++j) printf(" %d",v[i][j]+1);//存的时候少1，这里加回来
		puts("");
	}
	return 0;
}