P8536 「Wdoi-2」幻胧月睨 题解

Description

给你一个01字符串 \(s\) ， \(s_i=1\) 时， \(\forall 1\le j\le i-1,a_i>a_j\) , \(s_i =0\) 时， \(a_i\) 不满足上述条件，求任一满足条件的 \(a\) 数列。

Solution

设字符串 \(s\) 中有 \(x\) 个0, \(y\) 个1，则 \(\forall s_i=1,a_i=n-i+1\) ，即升序，既能保证 \(a_i>a_j\) ，也能保证 \(a_{i_1}<a_{i_2}\) ，再来看
\(s_i=0\) 的情况，明显剩下的 \(a_j=[1,x]\) ，但是为降序，否则当 \(s_1 =0,s_2=0\) 时， 可能\(a_1=1,a_2=2\)，实际上 \(s_1=0,s_2=1\) 造成冲突 。代码很好实现。

Code

int t;
int n;
int a[100100];
string s;
int main(){
    cin>>t;
    while(t--){
        int cnt=0,tot=0;
        cin>>n;
        cin>>s;
        for(int i=1;i<=n;i++){
            if(s[i-1]=='1') tot++;
          //  cout<<a[i];
        }
       	tot=n-tot+1; //不要忘了加一
        cnt=tot-1;
        for(int i=1;i<=n;i++){
            if(s[i-1]=='0'){
                cout<<cnt<<" ";
                cnt--;//递减
            }
            else{
                cout<<tot<<" ";
                tot++;//递增
            }
        }
        cout<<endl;
    }
    return 0;
}

完结撒花！！