剪枝
在深度优先搜索中 如果对所有的结果都尽心搜索是没有必要的 有的搜索并不是先要的结果 这时候就需要对它进行剪枝
如下图 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;
}
}