AGC029 A - Irreversible operation

問題はこちら atcoder.jp

方針

題意の操作は，文字列 においてBWをWBにすることと同じです。また， がWW...WBB...Bとなった時点で操作は終了します。

つまり，Wに注目すると， 回の操作で つのWが つ左に動き， 中にWが 個あるとき，一番うしろのWが 番目に来た時点で操作が終了します。

中で 個目のWが 番目にあるとき，これを 番目に移動させるためには　 回の操作が必要です。

コード

#include <bits/stdc++.h>
using namespace std;
 
int main(){
    string s;
    cin>>s;
 
    vector<int> coor_w;
    for(int i=0; i<s.size(); i++){
        if(s[i]=='W'){
            coor_w.push_back(i);
        }
    }
 
    long long count=0;
    for(int i=0; i<coor_w.size(); i++){
        count+=abs(coor_w[i]-i);
    }
 
    cout<<count<<endl;
}