1128 N Queens Puzzle

题目描述
思路： 在斜对角线就是x y的差值绝对值相同即可。

#include<iostream>
using namespace std;
int n,a[1010];
int main(){
    scanf("%d",&n);
    while(n--){
        int m;
        scanf("%d",&m);
        for(int i = 1;i <= m;i++)
            scanf("%d",&a[i]);
        bool not_solution = false;
        for(int i = 1;i <= m;i++){
            int y = i;
            int x = a[i];     
            for(int j = i+1;j <= m;j++){
                int sy = j;
                int sx = a[j];
                if(sy-y == 0 || sx-x == 0 || (abs(x-sx) == abs(y-sy))){
                    not_solution = true;
                    break;
                }
            }
            if(not_solution)
                break;
        }
        if(not_solution)
            cout<<"NO"<<endl;
        else cout<<"YES"<<endl;
    }
    return 0;
}