PAT 甲级 1148 Werewolf - Simple Version 20分

题目

题目描述

解析

由题意可知：说谎的两个人一个是狼，一个是村民。
所以可以枚举两只狼的位置，剩下的位置全是村民。
如果狼说某个人是狼但他却是村民或者说某个人是村民但他却是狼，那么说谎的狼的数量就要+1；如果村民说某个人是狼但他却是村民或者说某个人是村民但他却是狼，那么说谎的村民的数量就要+1。
最后如果说谎的狼和说谎的村民数量都为1，那么就直接输出并且结束程序，否则就输出No Solution。

代码

#include <cstdio>
#include <iostream>
#include <vector>

using namespace std;
 
int main()
{
	int n;
	cin >> n;
	vector<int> v(n + 1);
	for (int i = 1; i <= n; i++)
		cin >> v[i];
	for (int i = 1; i < n; i++){
		for (int j = i + 1; j <= n; j++){
			int wolf = 0, man = 0;
			for (int k = 1; k <= n; k++){
				if (k == i || k == j){
					//狼撒谎 
					if (v[k] > 0 && (v[k] == i || v[k] == j))
						wolf++;
					else if (v[k] < 0 && abs(v[k]) != i && abs(v[k]) != j)
						wolf++;
				}
				else{
					//村民撒谎 
					if (v[k] > 0 && (v[k] == i || v[k] == j))
						man++;
					else if (v[k] < 0 && abs(v[k]) != i && abs(v[k]) != j)
						man++;
				}
			}
			if (wolf == 1 && man == 1){
				cout << i << " " << j << endl;
				return 0;
			}
		}
	}
	cout << "No Solution" << endl;
	return 0;
}