1. 程式人生 > >【PAT乙級】1089 狼人殺-簡單版

【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;
}

總結

學習不息,繼續加油