ACM 荷蘭國旗問題

OpenSoucre發表於2014-04-01
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;
    }
}

 

相關文章