天梯赛题目L1-093 猜帽子游戏

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

输入格式：

输入首先在一行中给出一个正整数 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<bits/stdc++.h>
using namespace std;
int main()
{
	int n;
	cin>>n;
	int color_hat[n];
	//int color_hat[n] = {0};
	memset(color_hat,0,sizeof(color_hat));   //存储输入帽子颜色 
	for(int i=0;i<n;i++)
	{
		cin>>color_hat[i];   //循环输入帽子颜色  1黑2黄 
	}
	int cishu;
	cin>>cishu;
	for(int i=0;i<cishu;i++)   //控制判断次数 
	{
		int yes =0;
		int error=0;
		int a[n] = {0};
		//memset(a,0,sizeof(a));
		int sum=0;   //记录票值是否为全0   
		for(int j=0;j<n;j++)
		{
			cin>>a[j];    //循环输入游客的猜测  1黑2黄0弃权 
			sum+=a[j];    //将总和进行累加 
		}
		for(int j=0;j<n;j++)  //将color_hat里的元素与a中的元素进行比较 
		{
			if(sum==0)   //sum=0直接退出循环输出 
			{
				break;
			 } 
			else
			{
				if(a[j]==0) continue;    //a[j]等于0说明这段中有人弃权，则跳到下一个人 
				if(a[j]!=color_hat[j])  error++;
				if(a[j]==color_hat[j])  yes++;
			}
		}
		if(yes>=1 && error == 0) cout<<"Da Jiang!!!"<<endl;
		else
			cout<<"Ai Ya"<<endl;
	}
	return 0;
}