NYOJ_268_荷兰国旗问题

荷兰国旗问题

时间限制： 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

分析：荷兰国旗问题是一个经典的题目。因为只涉及到三种颜色，所以我们可以把红、白、蓝分别用1、2、3三个数字代替。可以把它看做是数组排序问题，而数组元素的值为1、2、3三种。排好序后数组肯定分为三块区域，前部为1，中部为2，后部为3。现在我们的思路就是把未排序时前部和后部分别排在数组的前面和后面，那么中部自然就排好了。

设置两个标志位head指向数组开头，tail指向数组末尾，now从头开始遍历：

(1)如果遍历到的位置为1，那么它一定是属于前部，于是就和head交换值，然后head++，now++；

(2)如果遍历到的位置为2，说明属于中部，now++；

(3)如果遍历到的位置为3，说明属于后部，于是就和tail交换值，然而，如果此时交换后now指向的值属于前部，那么就执行(1)，tail--；

其实这个题目如果只是解题的话完全不要这样子做，直接计算红、白、蓝三种颜色的数量，然后输出即可。

题目链接：http://acm.nyist.net/JudgeOnline/problem.php?pid=268

代码清单：

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

const int maxn = 100 + 5;

int n;
string str;
int main(){
    cin>>n;
    while(n--){
        cin>>str;
        int len=str.size();
        int now=0,ans=0;
        int head=0,tail=len-1;
        while(now<=tail){
            if(str[now]=='R'){
                swap(str[head],str[now]);
                head++;
                now++;
            }
            else if(str[now]=='W'){
                now++;
            }
            else{
                swap(str[now],str[tail]);
                tail--;
            }
        }
        cout<<str<<endl;
    }return 0;
}

#include<iostream>
using namespace std;
int main(){
    int n;
    cin>>n;
    string s="RWB";
    while(n--){
        string str;
        cin>>str;
        int ans[3]={0};
        int len=str.size();
        for(int i=0;i<len;i++){
            if(str[i]=='R') ans[0]++;
            else if(str[i]=='W') ans[1]++;
            else ans[2]++;
        }
        for(int i=0;i<3;i++){
            for(int j=0;j<ans[i];j++){
                cout<<s[i];
            }
        }cout<<endl;
    }return 0;
}