Day 30 代码随想录 N皇后

最新推荐文章于 2024-09-08 20:02:08 发布

别倒在黎明之前QAQ

最新推荐文章于 2024-09-08 20:02:08 发布

阅读量55

点赞数

分类专栏：算法训练营文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/m0_61375823/article/details/129298678

版权

算法训练营专栏收录该内容

47 篇文章 0 订阅

订阅专栏

本文介绍了使用Python的回溯算法解决经典的N皇后问题和数独求解问题。在N皇后问题中，重点在于is_valid函数的设计，而在解数独中，关键在于递归的backtracking函数和检查单元格有效性的方法。

摘要由CSDN通过智能技术生成

51. N 皇后

题目链接：51. N 皇后
在这里插入图片描述

这题想到回溯算法就不难，但是is_vailed函数 比较难想到。

class Solution:
    def __init__(self):
        self.res=[]
    def solveNQueens(self, n: int) :
        Chreerboard=[['.' for _ in range(n)]for _ in range(n) ]
        self.backtrACKING(Chreerboard,n,0)
        return self.res
    def backtrACKING(self,Chreerboard,n,row):
        if row==n:
            self.res.append([''.join(i) for  i  in Chreerboard])
            return
        #每一列中
        def isvaild(board,row,col):
            for i in range(len(board)):
                if board[i][col]=='Q':
                    return False
            # 排除自己本身
            i=row-1
            j=col-1
            # 排除45度
            while j>=0 and i>=0:
                if board[i][j]=='Q':
                    return False
                j-=1
                i-=1
            i=row-1
            j=col+1
            # 排除135度
            while i>=0 and j<len(board):
                if board[i][j] == 'Q':
                    return False
                i-=1
                j+=1
            return True
        #遍历每一列
        for col in range(n):
            if not isvaild(Chreerboard,row,col):
                continue
            Chreerboard[row][col]='Q'
            self.backtrACKING(Chreerboard,n,row+1)
            Chreerboard[row][col]='.'

37. 解数独

题目链接：37. 解数独
在这里插入图片描述

这题的思想还是要琢磨一下

class Solution:
    def solveSudoku(self, board) -> None:
        self.backtracking(board)
    def backtracking(self,board):
        for i in range (len(board)):
            for j in range(len(board[0])):
                if board[i][j] != '.': continue
                for k in range(1,10):
                    if self.isvaild(i,j,k,board):
                        board[i][j]=str(k)
                        if self.backtracking(board):return True
                        board[i][j]= '.'
                return False
        return True
    def isvaild(self,row,col,val,board):
        # 行
        for i in range(9):
            if board[row][i]==str(val):
                return False
        # 列
        for j in range(9):
            if board[j][col]==str(val):
                return False
        # 九宫格
        start_row=(row//3)*3
        start_col=(col//3)*3
        for i in range(start_row,start_row+3):
            for j in  range(start_col,start_col+3):
                if board[i][j]==str(val):
                    return False
        return True