AC代码

#include <bits/stdc++.h>
using namespace std;
int n;
set<int> s[2];
int main(){
cin>>n;
for(int i=1;i<=n;i++){
int t;
cin>>t;
s[t].insert(i);
}
while(s[0].size()||s[1].size()){
if(s[0].size()&&s[1].empty()){
for(auto i:s[0]){
cout<<i<<endl;
}
return 0;
}
if(s[1].size()&&s[0].empty()){
for(auto i:s[1]){
cout<<i<<endl;
}
return 0;
}
int t=*s[1].begin()<*s[0].begin();
int w=0;
while(1){
auto it=s[t].upper_bound(w);
if(it==s[t].end()){
break;
}
cout<<*it<<" ";
w=*it;
s[t].erase(it);
t=!t;
}
cout<<endl;
}
return 0;
}