PAT 甲级 1128. N Queens Puzzle (20) 【STL】

mac2022-06-30  20

题目链接

https://www.patest.cn/contests/pat-a-practise/1128

思路 可以 对每一个皇后 都判断一下 它的 行,列 ,左右对角线上 有没有皇后 深搜解决

但是这样太麻烦

其实我们可以想到

要符合要求的摆放 就是做到 每一行 每一列 每一条对角线 上 都是只有一个皇后的

每一行 就不用判断了

然后可以用 map 标记 该列 该对角线上 是否是已经有皇后的

如果已经有 那么就flag = 0

如果没有 就标记

然后 左对角线 上的特点就是 每个元素的 j - i 是相等的 右对角线上的特点是 i + j 是相等的

AC代码

#include <cstdio> #include <cstring> #include <ctype.h> #include <cstdlib> #include <cmath> #include <climits> #include <ctime> #include <iostream> #include <algorithm> #include <deque> #include <vector> #include <queue> #include <string> #include <map> #include <stack> #include <set> #include <numeric> #include <sstream> #include <iomanip> #include <limits> #define CLR(a) memset(a, 0, sizeof(a)) #define pb push_back using namespace std; typedef long long ll; typedef long double ld; typedef unsigned long long ull; typedef pair <int, int> pii; typedef pair <ll, ll> pll; typedef pair<string, int> psi; typedef pair<string, string> pss; const double PI = 3.14159265358979323846264338327; const double E = exp(1); const double eps = 1e-30; const int INF = 0x3f3f3f3f; const int maxn = 1e5 + 5; const int MOD = 1e9 + 7; int main() { int t; cin >> t; while (t--) { int n, num; int flag = 1; cin >> n; map <int, int> l, m, r; for (int i = 1; i <= n; i++) { scanf("%d", &num); if (m[num] || l[num - i] || r[num + i]) flag = 0; m[num] = 1; l[num - i] = 1; r[num + i] = 1; } if (flag) printf("YES\n"); else printf("NO\n"); } }

转载于:https://www.cnblogs.com/Dup4/p/9433199.html

相关资源:JAVA上百实例源码以及开源项目
最新回复(0)