ACM 荷兰国旗问题

荷兰国旗问题

时间限制：3000 ms  |  内存限制：65535 KB

难度：1

 

描述

 

荷兰国旗有三横条块构成，自上到下的三条块颜色依次为红、白、蓝。现有若干由红、白、蓝三种颜色的条块序列，要将它们重新排列使所有相同颜色的条块在一起。本问题要求将所有红色的条块放最左边、所有白色的条块放中间、所有蓝色的条块放最右边。

 

输入

第1行是一个正整数n（n<100），表示有n组测试数据。接下来有n行，每行有若干个由R，W，B三种字符构成的字符串序列，其中R，W和B分别表示红、白、蓝三种颜色的条块,每行最多有1000个字符。

输出

对输入中每行上由R，W，B三种字符构成的字符串序列，将它们重新排列使所有相同颜色的条块在一起，满足前述要求。

样例输入

3
BBRRWBWRRR
RRRWWRWRB
RBRW 

样例输出

RRRRRWWBBB
RRRRRWWWB
RRWB 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int main(){
    int n;
    cin >> n;
    for(int icase = 0 ; icase < n; ++ icase){
        string flag;
        cin >> flag;
        int leftIndex = 0,rightIndex = flag.length()-1;
        while(flag[leftIndex] == 'R') leftIndex++;
        while(flag[rightIndex] == 'B') rightIndex--;
        int currentIndex = leftIndex;
        while(currentIndex<= rightIndex){
            if(flag[currentIndex] == 'R') swap(flag[currentIndex++],flag[leftIndex++]);
            else if(flag[currentIndex] == 'B') swap(flag[currentIndex],flag[rightIndex--]);
            else currentIndex++;
        }
        cout<<flag<<endl;
    }
}