算法:N皇后问题

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

上图为 8 皇后问题的一种解法。

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

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

示例:

输入: 4
输出: [
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]
解释: 4 皇后问题存在两个不同的解法。
//N皇后:找到所有的解法。横着竖着两个方向的斜着都不能有皇后

var solutions [][]string

// n*n格的
func solveNQueens(n int) [][]string {
    solutions = [][]string{}

    queens := make([]int, n)
    for i := 0; i < n; i++ {
        queens[i] = -1
    }
    //列
    columns := map[int]bool{}

    diagonals1, diagonals2 := map[int]bool{}, map[int]bool{}
    //回溯
    backtrack(queens, n, 0, columns, diagonals1, diagonals2)
    return solutions
}


func backtrack(queens []int, n, row int, columns, diagonals1, diagonals2 map[int]bool) {
    if row == n {
        board := generateBoard(queens, n)
        solutions = append(solutions, board)
        return
    }

    //一列一列进行操作,固定列然后判断行
    for i := 0; i < n; i++ {
        //列
        if columns[i] {
            continue
        }

        diagonal1 := row - i
        if diagonals1[diagonal1] {
            continue
        }

        diagonal2 := row + i
        if diagonals2[diagonal2] {
            continue
        }

        queens[row] = i
        columns[i] = true
        diagonals1[diagonal1], diagonals2[diagonal2] = true, true
        //回溯
        backtrack(queens, n, row + 1, columns, diagonals1, diagonals2)
        queens[row] = -1

        delete(columns, i)
        delete(diagonals1, diagonal1)
        delete(diagonals2, diagonal2)
    }

}

func generateBoard(queens []int, n int) []string {
    board := []string{}
    for i := 0; i < n; i++ {
        row := make([]byte, n)
        for j := 0; j < n; j++ {
            row[j] = '.'
        }
        
[queens[i]] = 'Q'
        board = append(board, string(row))
    }
    return board
}

参考链接:https://leetcode-cn.com/problems/n-queens/solution/nhuang-hou-by-leetcode-solution/

  • 3
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

盼盼编程

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值