演算法 猜數字

_MaL發表於2020-10-16

小A 和 小B 在玩猜數字。小B 每次從 1, 2, 3 中隨機選擇一個,小A 每次也從 1, 2, 3 中選擇一個猜。他們一共進行三次這個遊戲,請返回 小A 猜對了幾次?

輸入的guess陣列為 小A 每次的猜測,answer陣列為 小B 每次的選擇。guess和answer的長度都等於3。

示例 1:

輸入:guess = [1,2,3], answer = [1,2,3]

輸出:3

解釋:小A 每次都猜對了。

示例 2:

輸入:guess = [2,2,3], answer = [3,2,1]

輸出:1

解釋:小A 只猜對了第二次。

演算法一:依次遍歷小A和小B的陣列,比較資料是否相同

int GuessNumber1(int arr[], int brr[])
{
	int count = 0;
	for (int i = 0;i < 3;i++)
	{
	if (arr[i] == brr[i])
			count++;
	}
}

演算法二:相對於演算法一的優化,使用vector陣列

int GuessNumber2(vector<int>& guess,vector<int>& answer)
{
	if (guess.size() != answer.size()) return 0;
	int cnt = 0;
	for (auto i: {0,1,2})		// i 取0,1,2
	{
		if (guess[i] == answer[i]) cnt++;
	}
	return cnt;
}

測試用例:

int main()
{
	vector<int> guess1 = { 1, 2, 3 };
	vector<int> answer1 = { 1, 2, 3 };
	auto ret1 = GuessNumber2(guess1, answer1);

	vector<int> guess2 = { 2,2,3 };
	vector<int> answer2 = { 3,2,1 };
	auto ret2 = GuessNumber2(guess2, answer2);


	cout << ret1 << "  " << ret2 << endl;

	return 0;
}

相關文章