pranay_teja
11/26/2018 - 5:48 PM

LeetCode-DIStringMatch

#include<bits/stdc++.h>
using namespace std;

vector<int> diStringMatch(string s) {
    int n=s.size();
    if(n==0){
        return {};
    }
    vector<int> ans(n+1);
    int start=0,end=n;
    int i=0;
    while(i<n){
        if(s[i]=='I'){
            ans[i]=start;
            start++;
        }else{
            ans[i]=end;
            end--;
        }
        i++;
    }
    ans[n]=start;
    return ans;
}
int main(){
    // freopen("ip.txt","r",stdin);
    // char ws;
    int t;
    cin>>t;
    // ws=cin.get();
    while(t--){
        string x;
        cin>>x;
        vector<int> a=diStringMatch(x);
        for(auto i: a){
            cout<<i<<" ";
        }
        cout<<endl;
    }
}