标签:return int self cols 34 II 算法 col row
算法 中等 | 34. N皇后问题 II
题目描述
根据n皇后问题,现在返回n皇后不同的解决方案的数量而不是具体的放置布局。
样例1
输入: n=1
输出: 1
解释:
1:
1
样例2
输入: n=4
输出: 2
解释:
1:
0 0 1 0
1 0 0 0
0 0 0 1
0 1 0 0
2:
0 1 0 0
0 0 0 1
1 0 0 0
0 0 1 0
java题解
dfs暴力搜索即可
public class Solution {
public static int sum;
public int totalNQueens(int n) {
sum = 0;
int[] usedColumns = new int[n];
placeQueen(usedColumns, 0);
return sum;
}
public void placeQueen(int[] usedColumns, int row) {
int n = usedColumns.length;
if (row == n) {
sum ++;
return;
}
for (int i = 0; i < n; i++) {
if (isValid(usedColumns, row, i)) {
usedColumns[row] = i;
placeQueen(usedColumns, row + 1);
}
}
}
public boolean isValid(int[] usedColumns, int row, int col) {
for (int i = 0; i < row; i++) {
if (usedColumns[i] == col) {
return false;
}
if ((row - i) == Math.abs(col-usedColumns[i])) {
return false;
}
}
return true;
}
}
C++题解
dfs暴力搜索即可
class Solution {
public:
int sum;
bool canPut(int row, int col, vector<int> &cols) {
for (int i = 0; i < row; i++) {
if (cols[i] - i == col - row) {
return false;
}
if (cols[i] + i == col + row) {
return false;
}
if (cols[i] == col) {
return false;
}
}
return true;
}
void dfs(int n, int k, vector<int> &cols) {
if (k == n) {
sum++;
return;
}
for (int i = 0; i < n; i++) {
if (!canPut(k, i, cols)) {
continue;
}
cols[k] = i;
dfs(n, k + 1, cols);
}
}
int totalNQueens(int n) {
vector<int> cols(n);
sum = 0;
dfs(n, 0, cols);
return sum;
}
};
python题解
dfs暴力搜索即可
class Solution:
total = 0
n = 0
def attack(self, row, col):
for c, r in self.cols.iteritems():
if c - r == col - row or c + r == col + row:
return True
return False
def search(self, row):
if row == self.n:
self.total += 1
return
for col in range(self.n):
if col in self.cols:
continue
if self.attack(row, col):
continue
self.cols[col] = row
self.search(row + 1)
del self.cols[col]
def totalNQueens(self, n):
self.n = n
self.cols = {}
self.search(0)
return self.total
CN_wanku
发布了157 篇原创文章 · 获赞 20 · 访问量 8508
私信
关注
标签:return,int,self,cols,34,II,算法,col,row 来源: https://blog.csdn.net/qq_43233085/article/details/104092122
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。