L1-5 猜帽子游戏

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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

栈限制

8192 KB

代码

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int N;
    cin >> N;
    vector<int> hat(N, -1);
    for (int &color : hat)
    {
        cin >> color;
    }
    int K;
    cin >> K;
    while (K--)
    {
        int rightNum = 0, wrongNum = 0;
        for (int i = 0; i < N; i++)
        {
            int guessColor;
            cin >> guessColor;
            if (guessColor == 0)
                continue;
            else if (guessColor == hat[i])
                rightNum++;
            else
                wrongNum++;
        }
        if (rightNum > 0 && wrongNum == 0)
            cout << "Da Jiang!!!\n";
        else
            cout << "Ai Ya\n";
    }
    return 0;
}

优化代码

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int N;
    cin >> N;
    vector<int> hat(N);
    for (int i = 0; i < N; i++)
    {
        cin >> hat[i];
    }
    int K;
    cin >> K;
    while (K--)
    {
        int rightNum = 0, wrongNum = 0;
        for (int i = 0; i < N; i++)
        {
            int guessColor;
            cin >> guessColor;
            if (guessColor == 0)
                continue;
            else if (guessColor == hat[i])
                rightNum++;
            else
                wrongNum++;
        }
        cout << ((rightNum > 0 && wrongNum == 0) ? "Da Jiang!!!\n" : "Ai Ya\n");
    }
    return 0;
}

• 移除了不必要的初始化。减少内存分配的开销和初始化时间。
• 使用三元运算符将输出逻辑合并到一行，使得代码更加简洁。