Codeforces-949A. Zebras

传送门

一个01串，问是否能分成k个子序列，使每个串始于0，终于0，且01交替出现。若可以，输出任意一种方案，否则输出-1

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long LL;

int len;
const int maxn = 2e5 + 10;
char s[maxn];
int nxt[maxn];
int bgn[maxn];

int main() {
    int ans= 0;
    scanf("%s", s + 1);
    len = strlen(s + 1);
    queue<int> q0, q1;
    for (int i = 1; i <= len; i++) {
        if (s[i] == '0') {
            if (q1.empty()) {
                q0.push(i);
                bgn[ans++] = i;
            } else {
                int tmp = q1.front(); 
                q1.pop();
                q0.push(i);
                nxt[tmp] = i;
            }
        } else {
            if (q0.empty()) {
                ans = -1;
                break;
            } else {
                int tmp = q0.front();
                q0.pop();
                q1.push(i);
                nxt[tmp] = i;
            }
        }
    }
    if (!q1.empty()) {
        puts("-1");
    } else {
        printf("%d
", ans);
        for (int i = 0; i < ans; i++) {
            int t = bgn[i];
            int cnt = 1;
            while (nxt[t]) {t = nxt[t]; cnt++;}
            t = bgn[i];
            printf("%d %d", cnt, t);
            while (nxt[t]) {
                printf(" %d", nxt[t]);
                t = nxt[t];
            }
            puts("");

        }
    }
    return 0;
}