PAT (Basic Level) Practice 1089

1089 狼人杀-简单版 （20 分）

以下文字摘自《灵机一动·好玩的数学》：“狼人杀”游戏分为狼人、好人两大阵营。在一局“狼人杀”游戏中，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

分析：题目有一句话很关键。“已知 N 名玩家中有 2 人扮演狼人角色，有 2 人说的不是实话，有狼人撒谎但并不是所有狼人都在撒谎。”说明1人为狼人说真话，1人为狼人说假话，1人为好人说假话。可以遍历所有的二人组合，若二人中一人为好人说假话，一人为狼人说真话，则可据此得到狼人的编号。

实现的方法是用一个二重循环，循环变量i, j为假设的狼人的编号，之后遍历整个玩家记录集合，若玩家的身份与其对应的话的结果不相同（例若其话为负，其话描述的人的身份为正），则此玩家说的是假话，将玩家编号记录下来。如果以此法得到的两个编号对应的玩家为1个狼人和1个好人，则i, j即为所求的编号。

代码：

#include<iostream>
#include<cmath>
#include<vector>
using namespace std;
int main() {
	int N;
	cin >> N;
	vector<int> v(N + 1);
	for (int i = 1; i <= N; i++) {
		cin >> v[i];
	}
	int flag = 0;
	for (int i = 1; i <= N; i++) {
		for (int j = i + 1; j <= N; j++) {
			vector<int> isWolf(N + 1, 1);
			isWolf[i] = isWolf[j] = -1;
			vector<int> lier;
			for (int k = 1; k <= N; k++) {
				if (v[k] * isWolf[abs(v[k])] < 0) {
					lier.push_back(k);
				}
			}
			if (lier.size() == 2 && isWolf[lier[0]] + isWolf[lier[1]] == 0) {
				cout << i << ' ' << j << endl;
				flag = 1;
				break;
			}
		}
		if (flag == 1) {
			break;
		}
	}
	if (flag == 0) {
		printf("No Solution\n");
	}
	return 0;
}