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

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

原题链接

AcWing 1621. N 皇后问题

算法标签

模拟 排序

思路

对棋盘的左右对角线，所在行分别开一个数组，然后判断左右对角线及所在行是否唯一即可

代码

#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总视频讲解

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈