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

题目链接

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
");
        else
            printf("NO
");
    }

}