【1128】N Queens Puzzle (20分)【逻辑题】

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>  
#include<map>
#include<vector>
#include<queue> 
using namespace std;  
//只是判断是否在同一行/对角线，题目隐含告诉不在同一列了！！
//vector（v），v[i]为该点在i列所在的行，像xy坐标函数。。
int main(){   
  int k,n;
  cin>>k;//k次查询
  for(int i=0;i<k;i++){
    cin>>n;
    vector<int>v(n);
    bool result=true;
    for(int j=0;j<n;j++){
      cin>>v[j];
      for(int t=0;t<j;t++){
        //下面if判断是否在同一行/同一对角线
        if(v[j]==v[t] || abs(v[j]-v[t])==abs(j-t)){
          result=false;
          break;
        }
      }
    }
    cout<<(result==true ?"YES\n":"NO\n");
  }
  system("pause");
    return 0;   
}