【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

分析
参考了柳婼的思路，才知道原来可以这么简单！

C++代码

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n;
    cin >> n;  //n个人
    vector<int> v(n+1);  //v[i]存第i个人说的话
    for(int i = 1; i <= n; i++)
        cin >> v[i];
    for(int i = 1; i <= n-1; i++) {
        for(int j = i+1; j <= n; j++) {
            vector<int> lie, a(n+1, 1);  //lie数组存说谎的人, a[i]存i是狼人(-1)或好人(1)
            a[i] = a[j] = -1;  //遍历两个人为狼人
            for(int k = 1; k <= n; k++) {  //判断所有人是否说谎,说谎则push_bakc进lie数组
                if(v[k] * a[abs(v[k])] < 0) lie.push_back(k);
            }
            if(lie.size()==2 && a[lie[0]]+a[lie[1]]==0) {  //判断当第i,j为狼人是是否符合题目条件
                cout << i << " " << j;
                return 0;
            }
        }
    }
    cout << "No Solution";
    return 0;
}