【PAT】1148 Werewolf - Simple Version

#include<bits/stdc++.h>
using namespace std;
vector<int> v(101);
int WH[101];//-1表示狼人，1表示人
int N;
//如果a,b,是狼人，判断是否符合要求，若符合要求就返回1，否则返回0；
int Judge(int a, int b) {
	//令a,b都是狼人，其他人都是人
	fill(WH, WH + 101, 1);
	WH[a] = WH[b] = -1;
	//接下来判断是否符合要求
	int sum1 = 0,sum2 = 0;//sum1表示狼人撒谎的人数，sum2表示人类撒谎的人数。
	for (int i = 1; i <= N; i++) {//遍历所有人找出每一个撒谎的人
		if (WH[abs(v[i])] * v[i] < 0) {
			if (WH[i] == -1) sum1++;
			if (WH[i] == 1) sum2++;
		}
		if (sum1 > 1 || sum2 > 1) return 0;//撒谎的人数量不符合要求
	}
	if (sum1 == 1 && sum2 == 1) return 1;
}

int main() {
#ifdef ONLINE_JUDGE
#else
	freopen("Text.txt", "r", stdin);
#endif // ONLINE_JUDGE
	scanf("%d", &N);
	for (int i = 1; i <= N; i++) {
		scanf("%d", &v[i]);
	}
	//直接上暴力，不怂它
	for (int i = 1; i < N; i++) {
		for (int j = i + 1; j <= N; j++) {
			if (Judge(i, j) == 1) {
				cout << i << " " << j;
				return 0;
			}
		}
	}
	cout << "No Solution";
}