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