第十一章模拟 20 AcWing 1621. N 皇后问题

淩亂°似流年 2024-03-31 13:30 33阅读 0赞

## 第十一章 模拟 20 AcWing 1621. N 皇后问题 ##

### 原题链接 ###

[AcWing 1621. N 皇后问题][AcWing 1621. N]

### 算法标签 ###

模拟 排序

### 思路 ###

对棋盘的左右对角线，所在行分别开一个数组，然后判断左右对角线及所在行是否唯一即可  
![在这里插入图片描述][9f766d315fe14cbe9070475c72c0c7c0.png]

### 代码 ###

#pragma GCC optimize(2)
    #pragma GCC optimize(3)
    #include<bits/stdc++.h>
    #define int long long
    #define xx first
    #define yy second
    #define ump unordered_map
    #define us unordered_set
    #define pq priority_queue
    #define rep(i, a, b) for(int i=a;i<b;++i)
    #define Rep(i, a, b) for(int i=a;i>=b;--i)
    using namespace std;
    typedef pair<int, int> PII;
    const int N=1005, inf=0x3f3f3f3f3f3f3f3f, mod=1e9+7;
    const double Exp=1e-8;
    //int t, n, m, cnt, ans; 
    int r[N], ed[N*2], ued[N*2];
    inline int rd(){
       int s=0,w=1;
       char ch=getchar();
       while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
       while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
       return s*w;
    }
    void put(int x) {
        if(x<0) putchar('-'),x=-x;
        if(x>=10) put(x/10);
        putchar(x%10^48);
    }
    signed main(){
    	ios::sync_with_stdio(false);
    	cin.tie(0);
    	cout.tie(0);
    	int t=rd();
    	while(t--){
    	    int n=rd();
    	    bool succ=true;
    	    memset(r, 0, sizeof r);
    	    memset(ed, 0, sizeof ed);
    	    memset(ued, 0, sizeof ued);
    	    rep(y, 1, n+1){
    	        int x=rd();
    	        if(r[x]||ed[x+y]||ued[y-x+n]){
    	            succ=false;
    	        }
    	        r[x]=ed[x+y]=ued[y-x+n]=true;
    	    }
    	    if(succ){
    	        puts("YES");
    	    }else{
    	        puts("NO");
    	    }
    	}
    	return 0;
    }

### 参考文献 ###

[AcWing 1621. N 皇后问题（PAT甲级辅导课）y总视频讲解][AcWing 1621. N _PAT_y]

原创不易  
转载请标明出处  
如果对你有所帮助 别忘啦点赞支持哈  
![在这里插入图片描述][9cb4cb80a18345158f8a69e2f196e5b9.png]

[AcWing 1621. N]: https://www.acwing.com/problem/content/1623/
[9f766d315fe14cbe9070475c72c0c7c0.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/6201b16c6774439ebfce046c56140e73.png
[AcWing 1621. N _PAT_y]: https://www.acwing.com/video/1281/
[9cb4cb80a18345158f8a69e2f196e5b9.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/31/091751db47ee466eaa6afd2102952475.png