leetcode第51题N 皇后

122 阅读1分钟

题目: 按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。

给你一个整数 n ,返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q' 和 '.' 分别代表了皇后和空位。 题目链接

我的JavaScript解法

/**
 * @param {number} n
 * @return {string[][]}
 */
var solveNQueens = function(n) {
  let chessboard = new Array(n).fill(new Array(n).fill('.').join(''));
  let result = [];
  backtrack(chessboard, 0, result)
  return result;
};

const backtrack = (chessboard, row, result) => {
  const len = chessboard.length;
  if (row == len) {
    result.push([...chessboard]);
    return;
  }
  for (let col = 0; col < len; col++) {
    if(!isValid(chessboard, row, col)) continue;
    let str = chessboard[row]
    chessboard[row] = str.substring(0, col) + 'Q' + str.substring(col + 1);
    backtrack(chessboard, row+1, result);
    chessboard[row] = str.substring(0, col) + '.' + str.substring(col + 1);
  }
}
const isValid = (chessboard, row, col) => {
  const len = chessboard.length;
  for (let i = 0; i <= row; i++) {
    if (chessboard[i].charAt(col) == 'Q')
      return false;
  }
  for (let i = row - 1, j = col + 1; 
    i >= 0 && j < len; i--, j++) {
    if (chessboard[i].charAt(j) == 'Q')
        return false;
  }
  for (let i = row - 1, j = col - 1;
    i >= 0 && j >= 0; i--, j--) {
    if (chessboard[i].charAt(j) == 'Q')
        return false;
  }
  return true;
}

**解析: ** 经典回溯算法,难点在于valid判断