PTA甲级考试真题练习148——1148 Werewolf - Simple Version

最新推荐文章于 2023-02-14 19:35:09 发布

醉等佳人归

最新推荐文章于 2023-02-14 19:35:09 发布

阅读量354

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT甲级考试真题练习

本文链接：https://blog.youkuaiyun.com/qq_43647628/article/details/105887433

PAT甲级考试真题练习专栏收录该内容

154 篇文章

订阅专栏

题目

在这里插入图片描述

思路

二重循环假设两个狼人，判断说谎人数，如果恰好为2并且一个是狼人一个是平民，则输出，若找不到则输出No Solution

代码

#include <iostream>
#include <vector>
#include <set>
using namespace std;
vector<int> vec;
int main()
{
	int n;
	cin >> n;
	vec.resize(n + 1);
	vector<int> mark;
	mark.resize(n + 1);
	fill(mark.begin(), mark.end(), 1);
	for (int i = 1; i <= n; ++i) cin >> vec[i];
	for (int i = 1; i <= n-1; ++i) {
		for (int j = i + 1; j <= n; ++j) {
			mark[i] = -mark[i];
			mark[j] = -mark[j];
			set<int> v;
			for(int k = 1; k <= n; ++k) {
				if (vec[k] * mark[abs(vec[k])] < 0)
					v.emplace(k);
			}
			if (v.size() == 2 && ((v.find(i) == v.end() && v.find(j) != v.end()) || (v.find(j) == v.end() && v.find(i) != v.end()))) {
				cout << i << " " << j;
				return 0;
			}
			mark[i] = -mark[i];
			mark[j] = -mark[j];
		}
	}
	cout << "No Solution";
	return 0;
}