fork download
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main(){
  5.  
  6. int size;
  7. cout<<"Total number of balls : ";
  8. cin>>size;
  9.  
  10. int ball[size];
  11. /// assumption red=0; blue=1; green=2;
  12.  
  13. int put[size];
  14. for(int i=0;i<size;i++){
  15.  
  16. cin>>ball[i];
  17. }
  18. int r=0;
  19. int g=size-1;
  20. int b=g-1;
  21.  
  22. for(int i=0;i<size;i++){
  23.  
  24. if(ball[i]==0) put[r++]=0;
  25. else if(ball[i]==2){
  26.  
  27. if(put[g]==1) put[b--]=1;
  28. put[g--]=2;
  29. }
  30. else put[b--]=1;
  31. }
  32. for(int i=0;i<size;i++) cout<<put[i]<<" "; cout<<endl;
  33. return 0;
  34. }
Success #stdin #stdout 0s 2688KB
stdin
6
1 0 2 0 1 2
stdout
Total number of balls : 0 0 1 1 2 2