1128 N Queens Puzzle (20分)

#include<iostream>
#include<algorithm>
#include<cstdio>
#define maxn 10005
using namespace std;

int main(){
   
	std::ios::sync_with_stdio(false);
	std::cin.tie(0);
	int k,n;
	cin>>k;
	while(k--){
   
		bool ans=true;
		cin>>n;
		int a[maxn];
		for(int i=1;i<=n;i++) cin>>a[i];
		for(int i=1;i<=n&&ans;i++){
   
			for(int j=1;j<i;j++)
			{
   
				if(a[i]==a[j]||abs(a[i]-a[j])==abs(i-j)){
   
					ans=false;
					break;
				}
			}
		}
		if(ans){
   
			cout<<"YES"<<endl;
		}
		else{
   
			cout<<"NO"<<endl;
		}
	}
}