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

结尾无空行

题目分析

用一个vector记录每个玩家的证词

然后从小到大枚举所有狼人存在的情况，对每一种情况，都遍历vector，如果满足有俩人撒谎还有一个是狼人，那就直接输出并return 0。

AC代码

#include <iostream>
#include <vector>
using namespace std;
vector<int> words;
int main(){int N;cin>>N;words.resize(N);for (int i = 0; i < N;i++){scanf("%d", &(words[i]));}for (int i = 1; i < N;i++){for (int j = i + 1; j <= N;j++){int liars = 0;//记录撒谎的人数int is_wolf = 0;//记录撒谎的狼人数for (int k = 0; k < words.size();k++){//遍历words数组，看看序号为k+1的狗东西有没有说谎if(words[k]<0)//说的是坏人{//又不是狼人if(words[k]!=-i&&words[k]!=-j){liars++;if(k+1 == i||k+1 == j)//撒谎的还是个狼人is_wolf++;}}else //说的是好人{//又是狼人if(words[k]==i||words[k]==j){liars++;if(k+1==i||k+1==j)//撒谎的还是个狼人is_wolf++;}}}if(liars==2&&is_wolf==1)//撒谎的有俩人还有一个是狼人！{cout << i << ' ' << j;return 0;}}}cout << "No Solution";return 0;
}

总结

判断撒谎人数那一部分肯定有好多可以优化的地方不过我好懒啊