<Sicily>Threecolor problem

一、题目描述

有红黄蓝3种颜色的n个珠子，师傅希望悟空把它们排成红色珠子在左，黄色珠子居中，蓝色珠子在右的一行，然后告诉师傅，从左数起，第m个珠子是什么颜色。众所周知，悟空是只猴子，他没有这个耐心，你帮帮他吧。

二、输入

输入第一行为一个整数t（0

三、输出

为每个测试用例单独一行输出指定的珠子颜色。
例如：
输入：
2
3 1
bry
4 3
rbyb
输出：
r
b

四、解题思路

这道题很水，只需要对输入的按照红、黄、蓝的次序进行排序，就可以了。这里使用的是插入排序（因为插入排序简单）。

五、代码

#include<iostream>

using namespace std;

int main()
{
    int times;
    cin >> times;

    while(times--)
    {
        int num, goal;
        cin >> num >> goal;

        char charArray[num];
        for(int i = 0; i < num; i++)
        {
            char newChar;
            cin >> newChar;
            charArray[i] = newChar;
        }

        for(int i = 1; i < num; i++)
        {
            int j = i - 1;
            char temp;
            temp = charArray[i];
            while(j >= 0)
            {
                if(charArray[j] == 'r') break;
                if(charArray[j] == 'y' && temp != 'r') break;
                if(charArray[j] == 'b' && temp == 'b') break;
                charArray[j+1] = charArray[j];
                j--;
            }
            charArray[j+ 1] = temp;
        }

        cout << charArray[goal - 1] << endl;
    }

    return 0;
}