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

题意：

输入一个正整数K（<=200），接着输入K行，第一个数为N（<=1000），后来跟着N个整数（1~N）表示第i列皇后的所在行数，所有的皇后不在一列，输出是否所有的皇后不会相互攻击。

trick：

N^2/2不会超时，N^2会在最后一个测试点超时。

AAAAAccepted code：

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
int a[1007];
int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int t;
    cin>>t;
    while(t--){
        int n;
        cin>>n;
        int flag=0;
        for(int i=1;i<=n;++i){
            cin>>a[i];
            for(int j=1;j<i;++j){
                int temp=i-j;
                int x=a[i]+temp;
                int y=a[i]-temp;
                if(a[j]==x||a[j]==y||a[j]==a[i])
                    flag=1;
            }
        }
        if(flag)
            cout<<"NO
";
        else
            cout<<"YES
";
    }
    return 0;
}