天梯赛L1-5猜帽子游戏

问题描述： 

宝宝们在一起玩一个猜帽子游戏。每人头上被扣了一顶帽子，有的是黑色的，有的是黄色的。每个人可以看到别人头上的帽子，但是看不到自己的。游戏开始后，每个人可以猜自己头上的帽子是什么颜色，或者可以弃权不猜。如果没有一个人猜错、并且至少有一个人猜对了，那么所有的宝宝共同获得一个大奖。如果所有人都不猜，或者只要有一个人猜错了，所有宝宝就都没有奖。
下面顺序给出一排帽子的颜色，假设每一群宝宝来玩的时候，都是按照这个顺序发帽子的。然后给出每一群宝宝们猜的结果，请你判断他们能不能得大奖。

输入格式：

输入首先在一行中给出一个正整数 N （ 2 < N ≤ 100 ），是帽子的个数。第二行给出 N顶帽子的颜色，数字 1 表示黑色，2 表示黄色。
再下面给出一个正整数 K （≤10），随后输入K 行，每行给出一群宝宝们猜的结果，除了仍然用数字 1 表示黑色、2 表示黄色之外，0 表示这个宝宝弃权不猜。同一行中的数字用空格分隔。

输出格式：

对于每一群玩游戏的宝宝，如果他们能获得大奖，就在一行中输出 Da Jiang!，否则输出 Ai Ya！。

输入样例：

5
1 1 2 1 2
3
0 1 2 0 0
0 0 0 0 0
1 2 2 0 2

​​​​​​​输出样例：

Da Jiang!
Ai Ya！
Ai Ya！

程序代码： 

#include<iostream>
#include<vector>
using namespace std;

int main() {
	int N, K;
	/*输入N顶帽子的颜色*/
	cin >> N;
	vector<int> arrN(N);//初始化一个长度为N的容器
	for (int i = 0; i < N; i++) {
		cin >> arrN[i];
	}

	/*输入K组猜帽子的结果*/
	cin >> K;
	vector < vector < int > >arrK(K,arrN);//初始化一个二维容器存放数组arrK[][], K行,与数组arrN[]列数保持一致;
	for (int i = 0; i < K; i++) {
		for (int j = 0; j < N; j++) {
			cin >> arrK[i][j];
		}
	}

	int tru, fal, qi;//tru表示猜对，fal表示猜错，qi表示弃权

	/*循环遍历K组结果，统计猜对、猜错和弃权的个数*/
	for (int i = 0; i < K; i++) {
		tru = 0; fal = 0; qi = 0;//每遍历一组结果之后，进行重置。
		for (int j = 0; j < N; j++) {
			if (arrK[i][j] == 0) {
				qi++;
			}
			else if (arrN[j] == arrK[i][j]) {
				tru++;
			}
			else {
				fal++;
			}
		}

		/*每统计一组结果，进行判断是否拿奖*/
		if (fal == 0 && tru > 0) {
			cout << "Da Jiang!" << endl;
		}
		else {
			cout << "Ai Ya!" << endl;
		}
	}
	return 0;
}