PAT 乙级练习 1089 狼人杀-简单版

PAT 乙级练习 题解合集

本题链接

题目

以下文字摘自《灵机一动·好玩的数学》：“狼人杀”游戏分为狼人、好人两大阵营。在一局“狼人杀”游戏中，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，2）（即假设 1 号和 2 号是狼人）开始遍历，考察这样的假设是否满足：有且仅有一个狼人说谎，同时有且仅有一个平民说谎。要是满足条件，就直接显示答案，如果不满足，继续遍历。

代码

#include <iostream>
#include <vector>
#include <utility>
#include <algorithm>
using namespace std;
int main() {
	int n, numInput, wolfA, wolfB;
	char symbolInput;
	cin >> n;
	vector<pair<int, bool> > input(n + 1); 
	vector<bool> assume(n + 1, true);
	for (int i = 1; i <= n; ++i) {
		getchar();
		cin >> symbolInput >> numInput;
		input[i] = {numInput, symbolInput == '+' ? true : false};
	}
	for (wolfA = 1; wolfA <= n - 1; ++wolfA) {
		for (wolfB = wolfA + 1; wolfB <= n; ++wolfB) {
			fill(assume.begin(), assume.end(), true); 
			assume[wolfA] = assume[wolfB] = false;
			if ((assume[input[wolfA].first] == input[wolfA].second && assume[input[wolfB].first] == input[wolfB].second) 
			 || (assume[input[wolfA].first] != input[wolfA].second && assume[input[wolfB].first] != input[wolfB].second))
				continue;
			int manLieCnt = 0;
			for (int i = 1; i <= n; ++i) {
				if (i == wolfA || i == wolfB)
					continue;
				if (assume[input[i].first] != input[i].second)
					++manLieCnt;
			}
			if (manLieCnt == 1)
				goto outside;
		} 
	}
outside:
	if (wolfA == n)
		puts("No Solution");
	else 
		cout << wolfA << " " << wolfB << endl;
	return 0;
}