Java实现N皇后问题回溯法求解

public class NQueens { private int[] queens; // 存储皇后所在列数的数组 private int n; // 皇后数量 private int count; // 解的数量

// 检查当前皇后放置的列是否合法
private boolean isValid(int row, int col) {
    for (int i = 0; i < row; i++) {
        if (queens[i] == col || Math.abs(row - i) == Math.abs(col - queens[i])) {
            return false;
        }
    }
    return true;
}

// 回溯搜索
private void search(int row) {
    if (row == n) { // 找到一个解
        count++;
        return;
    }

    for (int col = 0; col < n; col++) {
        if (isValid(row, col)) { // 当前皇后放置的列合法，则继续搜索下一行
            queens[row] = col;
            search(row + 1);
        }
    }
}

public int solveNQueens(int n) {
    this.n = n;
    queens = new int[n];
    count = 0;
    search(0);
    return count;
}

}