题解 | 小红的排列构造②

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n;cin>>n;
    string s;cin>>s;
    if(s.back()=='0'){
        cout<<-1<<'\n';
        return 0;
    }
    set <int> se;
    for(int i=1;i<=n;i++) se.insert(i);
    /*
    1 1 1 0 0 1 0 1
    1 2 3 5 6 4 8 7
    0 0 0 0 0 0 1 1
    2 3 4 5 6 7 1 8
    */
    for(int i=1;i<=s.size();i++){
        if(s[i-1]=='1'){
            cout<<*se.begin()<<' ';
            se.erase(se.begin());
            continue;
        }
        else{
            auto it=se.upper_bound(i);
            cout<<*it<<' ';
            se.erase(it);
            continue;
        }
    }
    return 0;
}
// 64 位输出请用 printf("%lld")