<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;
}