Bittipeli

input/code.cpp: In function 'int main()':
input/code.cpp:69:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for (int i=1; i<s.size(); ++i){
                  ^
input/code.cpp:78:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for (int i=0; i<l.size(); ++i){
                  ^
input/code.cpp:92:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if (mx>sm-mx-1-((mxi==k.size() || mxi==0 || (mxi==2 && k[0]==0))?0:1)){
                       ^
input/code.cpp:97:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (i=0; i<k.size(); ++i){
                ^
input/code.cpp:118:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i=0; i<ans.size(); ++i) cout << ans[i] << " "; cout << "\n";
                    ^

#include <iostream>
#include <vector>
using namespace std;
string s;

vector<int> l;
vector<int> k;
vector<int> ans;


vector<int> pp[2];
int ss[2];
int md, mg;


void pop(int r){
  if (r==1) ans.push_back(mg+md);
  else      ans.push_back(mg+1);
  
  if (md>1 || pp[!r].size()){
    --ss[r];
    --pp[r][pp[r].size()-1];  
  }
  if (md==1 && pp[!r].size()){
    --ss[!r]; 
    --pp[!r][pp[!r].size()-1];
  }else --md;
}

void mv(int r){
  if (r){
    mg-=pp[!r][pp[!r].size()-2];
  }else{
    mg+=md;
  }
  if (md || pp[r].size()) pp[r].push_back(md);
  pp[r].push_back(pp[!r][pp[!r].size()-1]);
  ss[r]+=pp[!r][pp[!r].size()-1]; 
  ss[!r]-=pp[!r][pp[!r].size()-1];
  pp[!r].pop_back();
  
  md=pp[!r][pp[!r].size()-1];
  pp[!r].pop_back();
}

void check(){
  for (int ri=0; ri<2; ++ri){
    if (pp[ri].size() && pp[ri][pp[ri].size()-1]==0){
      pp[ri].pop_back(); 
      if (pp[ri].size()){
        if (ri==0) mg-=pp[ri][pp[ri].size()-1]; 
        md+=pp[ri][pp[ri].size()-1]; 
        pp[ri].pop_back();
      }
    }
  }
  while (pp[1].size() && ss[1]-pp[1][pp[1].size()-1]>ss[0]) mv(0);
  while (pp[0].size() && ss[0]-pp[0][pp[0].size()-1]>ss[1]) mv(1);
  if (pp[1].size()==0 && pp[0].size()>0 && md==0) mv(1);
  if (pp[0].size()==0 && pp[1].size()>0 && md==0) mv(0);
}


int main(){
  ios_base::sync_with_stdio(0); cin.tie(0);
  cin >> s;
  
  int c=1;
  for (int i=1; i<s.size(); ++i){
    if (s[i]!=s[i-1]){l.push_back(c==1); c=0;}
    ++c;
  }l.push_back(c==1);
  
  int sm=0; 
  int mx=0; 
  int mxi=0;
  c=0; int ed=0;
  for (int i=0; i<l.size(); ++i){
    if (l[i]!=ed){
      k.push_back(c); 
      sm+=c;
      if (ed==1 && c>mx){ mx=c; mxi=k.size();}
      c=0; 
      ed=l[i];
    }
    ++c;
  }
  k.push_back(c); 
  sm+=c;
  if (ed==1 && c>mx){ mx=c; mxi=k.size();}
  
  if (mx>sm-mx-1-((mxi==k.size() || mxi==0 || (mxi==2 && k[0]==0))?0:1)){
   cout << "QAQ\n";
  }else{
    ss[0]=0;ss[1]=0;
    int i;
    for (i=0; i<k.size(); ++i){
      pp[0].push_back(k[i]);
      if (i&1) ss[0]+=k[i];
      else     mg+=k[i];
    }
    if (k.size()&1){
      mg-=k[--i];
      md=k[i];
      pp[0].pop_back();
    }else{
      md=0;
    }
    check();
    while (pp[0].size() || pp[1].size()){
      if (ss[0]>ss[1]) pop(0);
      else             pop(1);
      check();
      if (md<-1) return -1;
    }
    while (md>0){ ans.push_back(1); --md;}
    cout << ans.size() << "\n";
    for (int i=0; i<ans.size(); ++i) cout << ans[i] << " "; cout << "\n";
  }
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: ACCEPTED

Test 4

Group: 1

Verdict: ACCEPTED

Test 5

Group: 1

Verdict: ACCEPTED

Test 6

Group: 1

Verdict: ACCEPTED

Test 7

Group: 1

Verdict: ACCEPTED

Test 8

Group: 1

Verdict: ACCEPTED

Test 9

Group: 1

Verdict: ACCEPTED

Test 10

Group: 1

Verdict: ACCEPTED

Test 11

Group: 1

Verdict: ACCEPTED

Test 12

Group: 1

Verdict: ACCEPTED

Test 13

Group: 1

Verdict: ACCEPTED

Test 14

Group: 1

Verdict: ACCEPTED

Test 15

Group: 1

Verdict: ACCEPTED

Test 16

Group: 1

Verdict: ACCEPTED

Test 17

Group: 1

Verdict: ACCEPTED

Test 18

Group: 1

Verdict: ACCEPTED

Test 19

Group: 1

Verdict: ACCEPTED

Test 20

Group: 1

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: ACCEPTED

Test 22

Group: 2

Verdict: ACCEPTED

Test 23

Group: 2

Verdict: ACCEPTED

Test 24

Group: 2

Verdict: ACCEPTED

Test 25

Group: 2

Verdict: ACCEPTED

Test 26

Group: 2

Verdict: ACCEPTED

Test 27

Group: 2

Verdict: ACCEPTED

Test 28

Group: 2

Verdict: ACCEPTED

Test 29

Group: 2

Verdict: ACCEPTED

Test 30

Group: 2

Verdict: ACCEPTED

Test 31

Group: 2

Verdict: ACCEPTED

Test 32

Group: 2

Verdict: ACCEPTED

Test 33

Group: 2

Verdict: ACCEPTED

Test 34

Group: 2

Verdict: ACCEPTED

Test 35

Group: 2

Verdict: ACCEPTED

Test 36

Group: 2

Verdict: ACCEPTED

Test 37

Group: 2

Verdict: ACCEPTED

Test 38

Group: 2

Verdict: ACCEPTED

Test 39

Group: 2

Verdict: ACCEPTED

Test 40

Group: 2

Verdict: ACCEPTED

Test 41

Group: 3

Verdict: ACCEPTED

Test 42

Group: 3

Verdict: ACCEPTED

Test 43

Group: 3

Verdict: ACCEPTED

Test 44

Group: 3

Verdict: ACCEPTED

Test 45

Group: 3

Verdict: RUNTIME ERROR

Test 46

Group: 3

Verdict: ACCEPTED

Test 47

Group: 3

Verdict: ACCEPTED

Test 48

Group: 3

Verdict: ACCEPTED

Test 49

Group: 3

Verdict: ACCEPTED

Test 50

Group: 3

Verdict: ACCEPTED

Test 51

Group: 3

Verdict: ACCEPTED

Test 52

Group: 3

Verdict: ACCEPTED

Test 53

Group: 3

Verdict: ACCEPTED

Test 54

Group: 3

Verdict: RUNTIME ERROR

Test 55

Group: 3

Verdict: ACCEPTED

Test 56

Group: 3

Verdict: ACCEPTED

Test 57

Group: 3

Verdict: ACCEPTED

Test 58

Group: 3

Verdict: ACCEPTED

Test 59

Group: 3

Verdict: ACCEPTED

Test 60

Group: 3

Verdict: ACCEPTED

Test 61

Group: 4

Verdict: ACCEPTED

Test 62

Group: 4

Verdict: ACCEPTED

Test 63

Group: 4

Verdict: ACCEPTED

Test 64

Group: 4

Verdict: ACCEPTED

Test 65

Group: 4

Verdict: ACCEPTED

Test 66

Group: 4

Verdict: ACCEPTED

Test 67

Group: 4

Verdict: ACCEPTED

Test 68

Group: 4

Verdict: ACCEPTED

Test 69

Group: 4

Verdict: ACCEPTED

Test 70

Group: 4

Verdict: RUNTIME ERROR

Test 71

Group: 4

Verdict: ACCEPTED

Test 72

Group: 4

Verdict: ACCEPTED

Test 73

Group: 4

Verdict: ACCEPTED

Test 74

Group: 4

Verdict: ACCEPTED

Test 75

Group: 4

Verdict: ACCEPTED

Test 76

Group: 4

Verdict: ACCEPTED

Test 77

Group: 4

Verdict: ACCEPTED

Test 78

Group: 4

Verdict: ACCEPTED

Test 79

Group: 4

Verdict: ACCEPTED

Test 80

Group: 4

Verdict: ACCEPTED