PAT A1128

一看题，感觉是简单，就是写起来复杂啊，就在想可能有简单的方法，
没想到，开始动手写，AC是AC了，可是偶尔会超时
写完看柳神的代码，哎，我好菜啊

AC代码

#include<iostream>
#include<vector>
using namespace std;

int k, n, col;
int a[1010][1010] = {0};
int main() {
	cin >> k;
	while(k --) {
		vector<pair<int, int>> v;
		cin >> n;
		fill(a[0], a[0] + 1010 * 1010, 0);
		for(int row = 1; row <= n; row++) {
			cin >> col;
			a[row][col] = 1;
		}
		bool flag = true;
		for(int i = 1; i <= n; i++) {
			int ans = 0;
			for(int j = 1; j <= n; j++)
				if(a[i][j]) ans ++;
			if(ans >= 2) {
				flag = false;
				break;
			}
		}
		if(flag == false) {
			cout << "NO\n";
			continue;
		}
		for(int j = 1; j <= n; j++) {
			int ans = 0;
			for(int i = 1; i <= n; i++)
				if(a[i][j]) ans++;
			if(ans >= 2) {
				flag = false;
				break;
			}
		}
		if(flag == false) {
			cout << "NO\n";
			continue;
		}
		for(int i = 1; i <= n; i++)
			for(int j = 1; j <= n; j++)
				if(a[i][j]) v.push_back(make_pair(i, j));
		for(int i = 0; i < v.size(); i++) {
			for(int j = i + 1; j < v.size(); j++) {
				if(abs(v[i].first - v[j].first) == abs(v[i].second - v[j].second)) {
					flag = false;
					break;
				}
			}
			if(flag == false) break;
		}
		if(flag == false) cout << "NO\n";
		else cout << "YES\n";
	}
	return 0;
} 

柳神代码

#include<iostream>
#include<vector>
using namespace std;

int main() {
	int k, n;	
	cin >> k;
	while(k --) {
		cin >> n;
		vector<int> v(n + 1);
		bool result = true;
		for(int i = 1; i <= n; i++) {  // 依次读入第i列皇后处于的行号 
			cin >> v[i];
			for(int j = 1; j < i; j++) { // 如果第i列前有皇后和第i列皇后的行号相同或者处于对角线 ，输出NO 
				if(v[j] == v[i] || abs(v[j] - v[i]) == abs(j - i)) {
					result = false;
					break;
				}
			}
		}
		cout << (result ? "YES" : "NO") << endl;
	}
	return 0;
}