PAT 甲级 1128 N Queens Puzzle

题意：N皇后的问题，对于N个皇后而言，为了使所有的皇后不发生冲突，必须保证所有的皇后的行，列都不相等，而且任意两个皇后不位于同一对角线上面。

思路：用vector数组保存第j个数，下标j表示第j个数，值代表行数，当输入一个数之后，就与前面的所有数进行判断看是否存在冲突，如果输入数据之后然后在进行判断运行时间直接提升一倍，判断是否有位与一行的直接根据p[j]==p[k]判断，题目保证所有不位于一列，位于同一对角线的情况为：下标的绝对值只差==行数绝对值之差，即abs(j-k)==abs(p[j]==p[k])即认为位于同一对角线上

代码：

#include<iostream>
#include<vector>
using namespace std;
int main(){
	int n, m;
	cin >> n;
	for (int i = 0; i < n; i++) {
		int flag = 1;
		scanf("%d",&m);
		vector<int> p(m);
		for (int j = 0; j < m; j++) {//所有的行不相等，且不处于同一对角线(行的绝对值只差==下标的绝对值之差)
			scanf("%d",&p[j]);//输入数据
			for (int k = 0; k < j; k++) {
				if (p[j] == p[k] || abs(j - k) == abs(p[j] - p[k])) {
					flag = 0;
					break;
				}
			}
		}
		printf("%s\n",flag==1?"YES":"NO");
	}
	system("pause");
	return 0;
}