打印N-Queen问题中的所有解决方案

时间:2018-09-23 12:30:05

标签: javascript algorithm recursion

N Queen是在N×N棋盘上放置N个国际象棋皇后的问题,这样就不会有两个女王互相攻击。例如,下面是解决4 Queen问题的方法。

N Queen是在N×N棋盘上放置N个国际象棋皇后的问题,这样就不会有两个女王互相攻击。

我写了函数,但它只打印1个解决方案。如何更新此功能以打印所有解决方案?

Queen是“ q”,空值是“-”

function find_all_arrangements(n) {
    const allRes = [];

    function isValid(row, col, board) {
  // Checks the ← direction
    for(var i=0; i<col; i++){
      if (board[row][i] === "q") {
        return false;
      }
    }

    // Checks the ↖ direction 
    for(var i=row, j=col; i>=0 && j>=0; i--, j--){
      if (board[i][j] === "q") {
        return false;
      }
    }

    // Checks the ↙ direction 
    for(var i=row, j=col; j>=0 && i<n; i++, j--){
      if (board[i][j] === "q") {
        return false;
      }
    }

    return true;
    }

    function find(col, result) {
        if (col === n) {
            allRes.push(result);
      return true;
        }
        for (let i = 0; i < n; i++) {
            if (isValid(i, col, result)) {
                result[i][col] = "q";
                if (find(col + 1, result)) {
          return true;
        }
        result[i][col] = "-";
            }
        }
    return false;
    }

  function generateBoard(n){
    var board=[];
    for(var i=0; i<n; i++){
      board[i]=[];
      for(var j=0; j<n; j++){
        board[i][j]="-";
      }
    }
    return board;
  }

  var board = generateBoard(n);
  find(0, board);
  return allRes;
}

console.log(find_all_arrangements(4))

1 个答案:

答案 0 :(得分:3)

几乎在这里,只需修改find​​函数即可克隆结果数组并进行一些回溯调整:

function find(col, result) {
    if (col === n) {
      // this deep clone the 2d array
      allRes.push(JSON.parse(JSON.stringify(result)));
      return;
    }
    for (let i = 0; i < n; i++) {
        if (isValid(i, col, result)) {
            result[i][col] = "q";
            find(col + 1, result)
            result[i][col] = "-";
        }
    }
}