【PAT乙级】1089 狼人杀-简单版

以下文字摘自《灵机一动·好玩的数学》：“狼人杀”游戏分为狼人、好人两大阵营。在一局“狼人杀”游戏中，1 号玩家说：“2 号是狼人”，2 号玩家说：“3 号是好人”，3 号玩家说：“4 号是狼人”，4 号玩家说：“5 号是好人”，5 号玩家说：“4 号是好人”。已知这 5 名玩家中有 2 人扮演狼人角色，有 2 人说的不是实话，有狼人撒谎但并不是所有狼人都在撒谎。扮演狼人角色的是哪两号玩家？

本题是这个问题的升级版：已知 N 名玩家中有 2 人扮演狼人角色，有 2 人说的不是实话，有狼人撒谎但并不是所有狼人都在撒谎。要求你找出扮演狼人角色的是哪几号玩家？

输入格式：

输入在第一行中给出一个正整数 N（5≤N≤100）。随后 N 行，第 i 行给出第 i 号玩家说的话（1≤i≤N），即一个玩家编号，用正号表示好人，负号表示狼人。

输出格式：

如果有解，在一行中按递增顺序输出 2 个狼人的编号，其间以空格分隔，行首尾不得有多余空格。如果解不唯一，则输出最小序列解 —— 即对于两个序列 A=a[1],...,a[M] 和 B=b[1],...,b[M]，若存在 0≤k<M 使得 a[i]=b[i] （i≤k），且 a[k+1]<b[k+1]，则称序列 A 小于序列 B。若无解则输出 No Solution。

输入样例 1：

5
-2
+3
-4
+5
+4

输出样例 1：

1 4

输入样例 2：

6
+6
+3
+1
-5
-2
+4

输出样例 2（解不唯一）：

1 5

输入样例 3：

5
-2
-3
-4
-5
-1

输出样例 3：

No Solution

个人思路

这题在考试的时候也没做出来，今晚也花了好多时间，主要是一开始思路错了。一开始想遍历说谎的人的可能性，都写完了发现不可行。

正确的做法应该是遍历所有的狼人可能性，然后判断说谎个数，如果恰好有两人说谎且其中一个是狼人，一个是村民，则验证狼人成功。

有一个小坑，最后一个样例被卡了一会儿，就是字符串中的数字可能大于1位是多位数字，因此在从字符串转化成整型数字的时候要注意。

代码实现

#include <cstdio>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <vector>
#include <cmath>
#include <algorithm>
#include <iostream>
#define ll long long
#define ep 1e-5
#define INF 0x7FFFFFFF

const int maxn = 105;

using namespace std;

// records存着i号的发言
string records[maxn];
int n;

// 检查是否说谎
bool is_lie(string s, int status) {
    if ((s[0] == '-' && status == 1) || (s[0] == '+' && status == 0)) return false;
    else return true;
}

bool check(int wolf1, int wolf2) {
    // 假设wolf1和wolf2号是狼人，构造一个字符串存储狼人和村民信息
    int truth[maxn] = {0};
    truth[wolf1] = 1;
    truth[wolf2] = 1;
    
    // 检查是否说谎并统计个数
    int lie_cnt = 0, lie_wolf_num = 0;
    for (int i = 1; i <= n; i ++) {
        int i_target = 0;
        for (int j = 1; j < records[i].length(); j ++)
            i_target = i_target*10 + records[i][j]-'0';
        if (is_lie(records[i], truth[i_target])) {
            lie_cnt ++;
            if (i == wolf1 || i == wolf2) {
                lie_wolf_num ++;
            }
        }
    }
    
    // 返回结果
    if (lie_cnt == 2 && lie_wolf_num == 1) return true;
    else return false;
}

bool solve(int &wolf1, int &wolf2) {
    // 遍历所有的狼人可能性
    for (int i = 1; i <= n; i ++) {
        for (int j = i+1; j <= n; j ++) {
            // 验证狼人成功
            if (check(i, j)) {
                wolf1 = i;
                wolf2 = j;
                return true;
            }
        }
    }
    return false;
}

int main() {
    // 输入
    cin >> n;
    for (int i = 1; i <= n; i ++) {
        cin >> records[i];
    }
    
    // 寻找狼人并输出
    int wolf1 = 0, wolf2 = 0;
    if (!solve(wolf1, wolf2)) cout << "No Solution" << endl;
    else cout << wolf1 << " " << wolf2 << endl;
    
    return 0;
}

总结

学习不息，继续加油