L1-093 猜帽子游戏 Python

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

输入格式：

输入首先在一行中给出一个正整数 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

代码：

n=int(input())
ls1=list(map(int,input().split()))
m=int(input())
flag = 1
for _ in range(m):
    flag = 1
    ls2=list(map(int,input().split()))
    if ls2.count(0)==n:
        flag=0
    else:
        for i in range(n):
            if ls2[i]==ls1[i] or ls2[i]==0:
                pass
            else:
                flag=0
    if flag==1:
        print('Da Jiang!!!')
    else:
        print('Ai Ya')