PAT A1128

AC代码如下

#include <iostream>
#include <cmath>

using namespace std;

const int max_k=210;
int output[max_k]={0};
const int max_n=1100;
int queen[max_n]={0};

bool is_solution(int n,int a[]){
	for(int i = 0; i < n -1 ;i++){
		for(int j = i+1 ; j < n ;j++){
			if(a[j]==a[i]||(j-i)==abs(a[j]-a[i])) return false;
		}
	}
	return true;
}
int main(){
	int k,n;
	scanf("%d",&k);
	for(int i = 0 ; i < k;i++){
		scanf("%d",&n);
		for(int j = 0 ;j < n;j++){
			scanf("%d",&queen[j]);
		}
		if(is_solution(n,queen)){
			output[i]=1;
		}
	}
	for(int i = 0 ; i < k;i++){
		if(output[i]) printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}