Oleg writes down the history of the days he lived. For each day he decides if it was good or bad. Oleg calls a non-empty sequence of days a zebra, if it starts with a bad day, ends with a bad day, and good and bad days are alternating in it. Let us denote bad days as 0 and good days as 1. Then, for example, sequences of days 0, 010, 01010 are zebras, while sequences 1, 0110, 0101 are not.
Oleg tells you the story of days he lived in chronological order in form of string consisting of 0 and 1. Now you are interested if it is possible to divide Oleg‘s life history into several subsequences, each of which is a zebra, and the way it can be done. Each day must belong to exactly one of the subsequences. For each of the subsequences, days forming it must be ordered chronologically. Note that subsequence does not have to be a group of consecutive days.
In the only line of input data there is a non-empty string s consisting of characters 0 and 1, which describes the history of Oleg‘s life. Its length (denoted as |s|) does not exceed 200?000 characters.
If there is a way to divide history into zebra subsequences, in the first line of output you should print an integer k (1?≤?k?≤?|s|), the resulting number of subsequences. In the i-th of following k lines first print the integer li (1?≤?li?≤?|s|), which is the length of the i-th subsequence, and then li indices of days forming the subsequence. Indices must follow in ascending order. Days are numbered starting from 1. Each index from 1 to n must belong to exactly one subsequence. If there is no way to divide day history into zebra subsequences, print -1.
Subsequences may be printed in any order. If there are several solutions, you may print any of them. You do not have to minimize nor maximize the value of k.
0010100
3
3 1 3 4
3 2 5 6
1 7
111
-1
思路:
开始想用循环遍历所有的数组,将1插入到最后一位是0的后面,将0插入到1后面,如果无法插入1,那么就直接返回-1,但在第八组测试样例中我超时了,
于是看了学长的代码,他巧妙利用了一个index变量,将最后一位为0和最后一位是1的数组区分开,这样就免去了遍历的时间。而第八组数据全为0
总结:
#include <cstdio> #include <cstdlib> #include <iostream> #include <vector> #include <cstring> using namespace std; char words[200005]; vector<int>ans[200005]; int len,num=0,kk=0,index=0; int main() { scanf("%s",words); len=strlen(words); for(int i=0;i<len;i++) { if(words[i]==‘0‘)kk++; else kk--; } if(kk<1){cout<<-1<<endl;return 0;} for(int i=0;i<len;i++) { if(words[i]==‘0‘) { if(num<kk) { ans[num++].push_back(i); } else { if(--index<0){cout<<-1<<endl;return 0;} ans[index].push_back(i); } } else if(words[i]==‘1‘) { if(index+1>num){cout<<-1<<endl;return 0;} ans[index++].push_back(i); } } cout<<num<<endl; for(int i=0;i<num;i++) { printf("%d",ans[i].size()); for(int j=0;j<ans[i].size();j++) printf(" %d",ans[i][j]+1); puts(""); } return 0; }