数据结构极客视频5_DFS的题目

剪枝

在深度优先搜索中 如果对所有的结果都尽心搜索是没有必要的 有的搜索并不是先要的结果 这时候就需要对它进行剪枝
如下图 4皇后问题 虚线表示不必要的搜索

题目 N皇后问题

• 解题思路：对于每一行来说 如果在这行的第col列放了皇后 那么这列 以及它的撇和NA 和这一行都不能放。在pie和na中 是一个直线 x+y=c（常数），所以放皇后的时候 先判断能不能放 不能的话 就看看下一列 如果到了最后一层 row=n 那么结束 输出结果即可

class Solution:
    def solveNQueens(self, n: 'int') -> 'List[List[str]]':
        if n<1:
            return []
        self.result=[]
        self.pie=set()
        self.na=set()
        self.cols=set()
        self.dfs(n,0,[])
        return  self.generate(n)
    def dfs(self,n,row,result):
        if row>=n:
            self.result.append(result)
            return
        for col in range(n):
            if (col not in self.cols) and  (row-col not in self.pie) and (row+col not in self.na):## 相当于剪枝
                self.cols.add(col)
                self.pie.add(row-col)
                self.na.add(row+col)
                self.dfs(n,row+1,result+[col])
                self.cols.remove(col)## 换下一列了  上一列的皇后退出
                self.pie.remove(row-col)
                self.na.remove(row+col)
    def generate(self,n):
        ret=[]
        for i in self.result:
            for j in i:
                ret.append('.'*j + 'Q'+'.'*(n-j-1))
        return [ret[i:i+n] for i in range(0,len(ret),n)]

题目 数独 leetcode 36

解法：三个数组 用来记录该行 该列 该块 是否被占用 如果被占用 返回false 否则将该属加入该行 列 块

class Solution:
    def isValidSudoku(self, board: 'List[List[str]]') -> 'bool':
        row=[[ False for i in range(10)] for j in range(10)]        
        col=[[ False for i in range(10)] for  j in range(10)]
        block=[[ False for i in range(10)] for  j in range(10)]
        for i in range(9):
            for j in range(9):
                if board[i][j]=='.':
                    continue
                num = int(board[i][j])
                if(row[i][num] or col[j][num] or block[i//3*3+j//3][num]):
                    return False
                else :
                    row[i][num]=True
                    col[j][num]=True
                    block[i//3*3+j//3][num]=True
        return  True

题目 独数 leetcode 37

如果一个位置是空的 那么就尝试1-9的数 行不行遍历 如果空位置放1-9都不行 那么就是不行了 遍历后 没有false 那就true

class Solution {
    public void solveSudoku(char[][] board) {
        if (board==null || board.length==0){
            return ;
        }
        solve(board);
    }
    public boolean solve(char[][] board){
        for(int i=0;i<9;i++){
            for(int j=0;j<9;j++){
                if (board[i][j]=='.'){
                    for(char c='1';c<='9';c++){
                        if(valid(board,i,j,c)){
                            board[i][j]=c;                        
                            if(solve(board))
                               return true;                        
                            else
                               board[i][j]='.';
                        
                        }   
                    }
                    return false;                
            }
        }
       
    }
         return true;
    }
    private boolean valid(char[][] board,int i,int j,char c){
        for(int n=0;n<9;n++){
            if(board[i][n]!='.' && board[i][n]==c){
                return false;
            }
            if(board[n][j]!='.' && board[n][j]==c){
                return false;
            }
            if(board[i/3*3+n/3][j/3*3+n%3]!='.' && board[i/3*3+n/3][j/3*3+n%3]==c){
                return false;
            }
            
        }
        return true;
        
    }
}