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\n");

        else

            printf("NO\n");

    }

}

巴特西

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

最新文章

热门文章