PAT乙级1089(C++)——龙哥哥的刷题路

1089 狼人杀-简单版 (20分)
以下文字摘自《灵机一动·好玩的数学》：“狼人杀”游戏分为狼人、好人两大阵营。在一局“狼人杀”游戏中，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

假设i和j是狼人，接着遍历所有人说过的话，检测是否只有两个人撒谎，说谎的其中之一是狼人。如果是的话，就输出结果，退出循环；如果不是继续循环，直到有解或者遍历结束

#include<iostream>
#include<string>
#include<set>
using namespace std;
int main(){
	int N = 0;
	cin >> N;
	int a[105];
	bool flag = 0,flag2 = 0;
	for (int i = 1; i <= N; i++){
		cin >> a[i];
	}
	for (int i = 1; i < N && !flag2; i++){
		for (int j = i+1; j <= N && !flag2; j++){
			int count = 0;
			int b[3] = { 0, 0, 0 };
			for (int k = 1; k <= N; k++){
				if (a[k]>0){
					if (a[k]==i || a[k]==j){
						b[count++]=k;
						if (count >= 3){
							break;
						}
					}
				}
				else{
					if (a[k]!=-i && a[k]!=-j){
						b[count++]=k;
						if (count >= 3){
							break;
						}
					}
				}
			}
			if (count==2 && ((b[0]==i || b[0]==j || b[1]==i || b[1]==j) && (b[0]+b[1]!=i+j))){
				flag2 = 1;
				printf("%d %d",i,j);
				break;
			}
		}
	}
	if (!flag2){
		printf("No Solution");
	}
	return 0;
}