【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;
}
總結
學習不息,繼續加油