leetcode51-N皇后

leetcode 51

思路

本题可以使用回溯算法来解决。回溯算法通过尝试所有可能的解决方案来找到问题的解的算法,当发现当前的选择无法得到有效的解决方案时,就回溯到上一步,尝试其他的选择。对于 N 皇后问题,我们可以逐行放置皇后,在每一行选择一个合适的列来放置皇后,若当前选择导致冲突,就回溯到上一行,重新选择列

初始化棋盘
js 复制代码
const dashboard = Array(n).fill().map(() => Array(n).fill('.'));
  • dashboard 是一个 n×n 的二维数组,初始时每个位置都用 . 表示,表示该位置没有放置皇后
回溯函数
js 复制代码
const backtracking = (dashboard, n, row) => {
  if (row === n) {
    result.push(dashboard.map(item => item.join('')))
    return;
  }
  for (let i = 0; i < n; i++) {
    if (isValid(dashboard, row, i, n)) {
      dashboard[row][i] = 'Q'
      backtracking(dashboard, n, row + 1)
      dashboard[row][i] = '.'
    }
  }
}
  • 终止条件:当 row 等于 n 时,说明已经成功地在每一行都放置了一个皇后,此时将当前的棋盘布局加入到 result 数组中
  • 遍历列:对于当前行 row,尝试在每一列 i 放置皇后
  • 检查合法性:调用 isValid 函数检查在 (row, i) 位置放置皇后是否合法
  • 递归:如果合法,将皇后放置在 (row, i) 位置,然后递归调用 backtracking 函数处理下一行
  • 回溯:递归返回后,将 (row, i) 位置的皇后移除,恢复为 .,以便尝试其他列
棋盘的合法性校验:关键!!!
js 复制代码
const isValid = (dashboard, row, col, n) => {
  if (row === 0) { return true }
  // 判断是否在同一列
  for (let i = 0; i < row; i++) {
    if (dashboard[i][col] === 'Q') {
      return false
    }
  }
  // 判断是否在45度角
  for (let i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
    if (dashboard[i][j] === 'Q') {
      return false
    }
  }
  // 判断是否是135度角
  for (let i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
    if (dashboard[i][j] === 'Q') {
      return false
    }
  }
  return true
}
  • 第一行:如果 row 为 0,说明是第一行,任何位置都可以放置皇后,直接返回 true
  • 列检查:检查当前列 col 上是否已经有皇后,如果有则返回 false
  • 45 度斜线检查:从当前位置 (row, col) 向左上方遍历,如果发现有皇后则返回 false
  • 135 度斜线检查:从当前位置 (row, col) 向右上方遍历,如果发现有皇后则返回 false
  • 合法:如果以上检查都通过,则返回 true,表示该位置可以放置皇后

完整实现

js 复制代码
var solveNQueens = function (n) {
  let result = [];
  // 初始化棋盘
  const dashboard = Array(n).fill().map(() => Array(n).fill('.'));
  const backtracking = (dashboard, n, row) => {
    if (row === n) {
      result.push(dashboard.map(item => item.join('')))
      return;
    }
    for (let i = 0; i < n; i++) {
      if (isValid(dashboard, row, i, n)) {
        dashboard[row][i] = 'Q'
        backtracking(dashboard, n, row + 1)
        dashboard[row][i] = '.'
      }
    }
  }
  backtracking(dashboard, n, 0)
  return result;
}

const isValid = (dashboard, row, col, n) => {
  if (row === 0) { return true }
  // 判断是否在同一列
  for (let i = 0; i < row; i++) {
    if (dashboard[i][col] === 'Q') {
      return false
    }
  }
  // 判断是否在45度角
  for (let i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
    if (dashboard[i][j] === 'Q') {
      return false
    }
  }
  // 判断是否是135度角
  for (let i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
    if (dashboard[i][j] === 'Q') {
      return false
    }
  }
  return true
}
相关推荐
virtual_k1smet2 小时前
#等价于e * d ≡ 1 mod φ(n) #模逆元详解
人工智能·算法·机器学习
可触的未来,发芽的智生3 小时前
新奇特:神经网络的集团作战思维,权重共享层的智慧
人工智能·python·神经网络·算法·架构
_屈臣_3 小时前
卡特兰数【模板】(四个公式模板)
c++·算法
坚持编程的菜鸟3 小时前
LeetCode每日一题——交替合并字符串
c语言·算法·leetcode
悦悦子a啊3 小时前
[Java]PTA: jmu-Java-02基本语法-08-ArrayList入门
java·开发语言·算法
xlq223224 小时前
12.排序(上)
数据结构·算法·排序算法
努力学习的小廉5 小时前
我爱学算法之—— 分治-快排
c++·算法
未知陨落5 小时前
LeetCode:77.买卖股票的最佳时机
算法·leetcode
DeeGLMath5 小时前
排序算法的并行加速实现
算法·排序算法·joblib·sortingx
光影少年5 小时前
vue打包优化方案都有哪些?
前端·javascript·vue.js