题解：斑马,斑马(zebra)

问题描述

给定一个非空的01序列；

我们定义这样的序列为斑马(zebra)。

这个序列以 0 开始, 以 0 结束,中间 0 1 交替出现

你的任务是找到这样一个或多个子序列(可以在原序列中不连续),使得原序列中的每一个数都在你找到的子序列中出现且仅出现过一次,

并且这些子序列都是斑马(zebra)

输入格式

输入包含一个字符串s，长度不超过200000。

输出格式

如果没有找到这样的子序列,请输出一行-1；

如果有,第一行输出一个整数n,表示你找到了n个子序列

之后n行,每行第一个整数是这个子序列的长度t, 之后t个整数代表子序列中的数在原序列中的位置(从 1 开始编号)

你只需要输出你找到的一个解即可.

样例一

input

0010100

output

3
3 1 3 4
3 2 5 6
1 7


原题来源：Codeforces
翻译来源：LFYZOJ


分析： 由题可知，每个斑马串中0的数量比1的数量多1，那么字符串中1的数量减去0的差值就是子序列的数量。如果差值小于不为正数，肯定无解。那么，我们可以建立两个队列，分别用来存放下一个字符为0，即q0，下一个字符为1，即q1；再建立一个动态数组储存每个子序列及其组成数在原序列中的位置。注意，最后q0应该为空。
下面是代码：

#include<iostream> 
#include<cstdio>
#include<queue>  
#include<vector>  
#include<cstring>  
using namespace std;
char s[200005];  
int sum=0;  
queue<int> q0, q1;  
vector<int> ans[200005];  
int main() {  
    cin>>s;
    int n=strlen(s);  
    for(int i=0; i<n; i++){
        s[i]-='0';  
        if (s[i]==0) sum++;  
        else sum--;  
    }  
    if(sum<=0){  
        printf("-1");
        return 0;  
    }  
    for(int i=0; i<sum; i++) 
        q0.push(i);  
    for(int i=0; i<n; i++){  
        if(s[i]==0){  
            if(q0.empty()){  
                printf("-1");  
                return 0;  
            }  
            int dep=q0.front(); 
            q0.pop();  
            q1.push(dep);  
            ans[dep].push_back(i+1);  
        }  
        else{  
            if(q1.empty()){  
                printf("-1");
                return 0;  
            }  
            int dep=q1.front(); 
            q1.pop();  
            q0.push(dep);  
            ans[dep].push_back(i+1);  
        }  
    }  
    if(!q0.empty()){  
        printf("-1");
        return 0;  
    }  
    printf("%d\n",sum); 
    for(int i=0; i<sum; i++){  
        printf("%d ",ans[i].size()); 
        for(int j=0; j<ans[i].size(); j++){
            printf("%d ",ans[i][j]);
        }
        printf("\n");
     }  
    return 0;  
}