PAT A 1148 Werewolf - Simple Version (20 分)

一、题目概述

狼人杀游戏中，有村民和狼人两种身份。N个玩家中共有两个狼人，有两个人说谎且有一个是狼人。要求找出游戏中的两只狼。

二、思路

每次迭代假设两位玩家是狼人，检查每位玩家的发言，若玩家发言所说的身份和对方被假设的身份不符，则假设该玩家说谎。若该轮迭代共两人说谎，且其一为狼人，则为所求。

迭代假设从低序号开始，因此找到的第一对狼就是最小解。

三、代码

#include <cstdio>
#include <vector>
using namespace std;
int main()
{
    int N;
    scanf("%d", &N);
    vector<int> statement(N + 1);
    for( int i = 1; i <= N; ++i )
        scanf("%d", &statement[i]);
    for( int i = 1; i < N; ++i )
        for( int j = i + 1; j <= N; ++j )
        {
            int cnt = 0;
            vector<bool> lied(N + 1, false), iswolf(N + 1, false);
            iswolf[i] = iswolf[j] = true;
            for( int k = 1; k <= N; ++k )
                if( ( statement[k] > 0 && iswolf[ statement[k] ] ) || ( statement[k] < 0 && !iswolf[ statement[k] * -1 ] ) )
                {
                    ++cnt;
                    lied[k] = 1;
                }
            if( cnt == 2 && lied[i] + lied[j] == true )
            {
                printf("%d %d", i, j);
                return 0;
            }
        }
    printf("No Solution");
}