C++ 代码

#include <iostream>
using namespace std;

const int MAX_N = 1010;
int q[MAX_N];

bool check(int n) {
    for (int i = 1; i <= n; i++) {
        for (int j = i + 1; j <= n; j++) {
            //检查行和对角线
            if (q[i] == q[j] || abs(i - j) == abs(q[i] - q[j])) {
                return false;
            }
        }
    }
    return true;
}

int main() {
    int n, m;
    scanf("%d", &n);
    for (int i = 0; i < n; i++) {
        scanf("%d", &m);
        for (int j = 1; j <= m; j++) {
            scanf("%d", &q[j]);
        }
        if (check(m)) puts("YES");
        else puts("NO");
    }
    return 0;
}