CF949A Zebras

发布时间 2023-08-23 14:44:33作者: One_JuRuo

思路

我们可以先模拟一下:

如果发现该位为 \(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;
}