Zebras CodeForces - 950C

点击打开链接

遇到一个0 先看有没有以1结尾的串 有就补上 没有就新开一个串

遇到一个1 有以0结尾的串就填上 没有就输出-1

如果每次都遍历所有已经构造的串 复杂度太高 可以用队列优化

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

vector <int> ans[200010];
queue <int> que0,que1;
int n,k;
char ch[200010];

int main()
{
    int i,j,p,flag;
    scanf("%s",ch);
    n=strlen(ch),k=0,flag=1;
    for(i=0;i<n;i++)
    {
        if(ch[i]=='0')
        {
            if(que1.empty())
            {
                k++;
                ans[k].push_back(i+1);
                que0.push(k);
            }
            else
            {
                p=que1.front();
                que1.pop();
                ans[p].push_back(i+1);
                que0.push(p);
            }
        }
        else
        {
            if(que0.empty())
            {
                flag=0;
                break;
            }
            else
            {
                p=que0.front();
                que0.pop();
                ans[p].push_back(i+1);
                que1.push(p);
            }
        }
    }
    if(!flag||!que1.empty())
    {
        printf("-1\n");
    }
    else
    {
        printf("%d\n",k);
        for(i=1;i<=k;i++)
        {
            printf("%d ",ans[i].size());
            for(j=0;j<ans[i].size();j++)
            {
                printf("%d ",ans[i][j]);
            }
            printf("\n");
        }
    }
    return 0;
}