1884B - Haunted House - CodeForces Solution

#include <bits/stdc++.h>
using namespace std;
int solve(int t,string str){

}
int main(){
    int t;
    cin>>t;
    while(t--){
        int n;
        string str;
        cin>>n>>str;
        int n0=0;
        queue<long long> s1;
        for(int i=n-1;i>=0;i--){
            if(str[i]=='0'){
                n0++;
                s1.push(i);
            }
        }
        long long ans=0;
        for(int i=1;i<=n0;i++){
            int idx=n-i;
            ans+=idx-s1.front();
            cout<<ans<<" ";
            s1.pop();
        }
        for(int i=n0+1;i<=n;i++){
            cout<<"-1 ";
        }
        cout<<endl;
    }
}