题目 P7912 [CSP-J 2021] 小熊的果篮

#include<bits/stdc++.h>
using namespace std;
int n,i,p,s,a[350001],l=1;
bool b[350001];
int main()
{
	memset(b,true,sizeof(b));
	scanf("%d",&n);
	for(i=1;i<=n;++i) scanf("%d",&a[i]);
	p=n;
	s=a[1];
	while(p>0)
	{
		for(i=1;i<=n;++i)
		if(b[i]==true)
		{
			s=a[i];
			l=i;
			if(s==1) s=0;
			else s=1;
			break;
		}
		for(i=l;i<=n;++i)
		{
			if(a[i]!=s&&b[i]==true)
			{
				printf("%d ",i);
				s=a[i];
				p--;
				b[i]=false;
			}
		}
		cout<<endl;
	}
	return 0;
}