Pat乙级1089 狼人杀-简单版

题目网址
https://pintia.cn/problem-sets/994805260223102976/problems/1038429385296453632

思路

题目中一共两个狼人，其中只有一个狼人说谎，好人阵营中一个好人说谎，一共两个人说谎。用两层for循环假设狼人身份，再内嵌一层循环去找说谎的好人。找到说谎的好人数量必须为1，这时候验证这两个狼人是不是只有一个人在说谎，找到就输出，结束程序，遍历完还找不到就是没有。

代码

#include<iostream>
#define MAX 105

using namespace std;

int A[MAX];

int fun(int i, int j, int a) {
	if (a < 0 && (-a) != i && (-a) != j)
		return 1;
	if (a > 0 && (a == i || a == j))
		return 1;
	return 0;
}

int main() {
	int n;
	int cnt;
	int sum;
	
	cin >> n;
	for(int i = 1; i <= n; i ++)
		cin >> A[i];
	for(int i = 1; i <= n; i ++) {
		for (int j = i + 1; j <= n; j ++) {
			cnt = 0;
			sum = 0;
			for (int k = 1; k <= n; k ++) {
				if (k != i && k != j) {
					if (A[k] > 0) {
						if (A[k] == i || A[k] == j)
							cnt ++;
					}
					else
						if ((-A[k]) != i && (-A[k]) != j)
							cnt ++;
				}
			}
			if (cnt == 1) {
				sum += fun(i, j, A[i]);
				sum += fun(i, j, A[j]);
				if (sum == 1) {
					cout << i << " " << j << endl;
					return 0;
				}
			}
		}
	}
	cout << "No Solution" << endl;
	return 0;
}