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

最新推荐文章于 2020-07-30 21:10:13 发布

Dup4

最新推荐文章于 2020-07-30 21:10:13 发布

阅读量165

点赞数

分类专栏： PAT甲级文章标签： STL

本文链接：https://blog.csdn.net/Dup4plz/article/details/79716158

版权

PAT甲级专栏收录该内容

8 篇文章 0 订阅

订阅专栏

题目链接

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");
    }

}