面试题 猜颜色球游戏

游戏规则：

有四个插口， 可以放有四个球，颜色分别为红色（R），黄色（Y），绿色（G）， 和蓝色（B），放的颜色顺序位置都可以是随机的，如：RYGB,YGRB等都是合法放置。

我们需要猜测四个颜色，比如原插口放置球是RGBY;我们猜测RRGG，那么第一个R正好对应我们有一个"hit"，猜了个G，但是位置没对应，就有一个"pseudo-hit".而且不能算两个G，因为原插口只有一个G。

要我们设计一个算法可以返回hits和pseudo-hits的数量。

注意：插口有RGBY，猜GGRR只能是一个hit和一个pseudo hit。原书的程序应该是有点错误的，按他的程序这个解答就是1个hit和3个pseudo hit了。

先吃透他的规则，注意特殊情况，然后就好做了。

书中提到的注意地方：

1 代码清晰，比如定义一个结构体返回结果

2 定义额外函数

这里用到的一个最基础的知识就是计算排序counting sort，这个算法很重要，好像出现的地方很多，就算不是直接使用counting sort，类似的使用方法经常出现，比如这里也用到了相关知识。计数算法参考：http://blog.csdn.net/kenden23/article/details/12437455

程序：

#include<iostream>

using namespace std;

const int SLOTSCOLORS = 4;

int rgbyToIndex(char c)
{
	switch (c)
	{
	case 'R':
		return 0;
		break;
	case 'G':
		return 1;
		break;
	case 'B':
		return 2;
		break;
	case 'Y':
		return 3;
		break;
	default:
		return -1;
		break;
	}
}

struct Result
{
	int hits;
	int pseudoHits;
	Result():hits(0), pseudoHits(0){}
};

Result guessHits(char *slots, char *guess)
{
	if (slots == nullptr || guess == nullptr)
	{
		return Result();
	}
	Result res;
	int counting[SLOTSCOLORS];
	for (int i = 0; i < SLOTSCOLORS; i++)
	{
		if(guess[i] == slots[i])
		{
			res.hits ++;
			//注意：要把数组加标志-1，标明是hit，就不能计算入pseudo-hit了。
			counting[rgbyToIndex(guess[i])] = -1;
		}
		else if (counting[rgbyToIndex(guess[i])] != -1)
		{
			counting[rgbyToIndex(guess[i])] = 1;
		}
	}
	for (int i = 0; i < SLOTSCOLORS; i++)
	{
		if(counting[i] == 1)
			res.pseudoHits ++;
	}
	return res;
}

int main() 
{
	char slots[4] = {'R','G','B','Y'};
	char guess[4] = {'G','G','R','R'};
	Result res = guessHits(slots, guess);
	cout<<"hits : pseudo hits\n";
	cout<<res.hits<<" : "<<res.pseudoHits<<endl;

	system("pause");
	return 0;
}

运算结果：

 

Reference:

Cracking the coding interview