[Leetcode]N-Queens II
2015-12-15 17:22
441 查看
Follow up for N-Queens problem.
Now, instead outputting board configurations, return the total number of distinct solutions.
class Solution {
public:
/*algorithm: DFS
*/
vector<int>getNextCandiates(vector<int>&solution,int n)
{
vector<int>result;
for(int i = 0; i < n;i++){
bool selected = true;
for(int k = 0;k < solution.size();k++){
if(i == solution[k] || //same column or same k=1/-1 (k=(y2-y1)/(x2-x1))
abs(i-solution[k])==abs((int)solution.size()-k)){
selected = false;
break;
}
}
if(selected)result.push_back(i);
}
return result;
}
void dfs(int n,int start,int &sum,vector<int>&solution){
if(start == n){ //find one solution,reach last line
sum++;
return;
}
vector<int>candiates = getNextCandiates(solution,n);
for(int k = 0;k < candiates.size();k++){
solution.push_back(candiates[k]);
dfs(n,start+1,sum,solution);
solution.pop_back();
}
}
int totalNQueens(int n) {
vector<int>solution;
int sum = 0;
dfs(n,0,sum,solution);
return sum;
}
};
Now, instead outputting board configurations, return the total number of distinct solutions.
class Solution {
public:
/*algorithm: DFS
*/
vector<int>getNextCandiates(vector<int>&solution,int n)
{
vector<int>result;
for(int i = 0; i < n;i++){
bool selected = true;
for(int k = 0;k < solution.size();k++){
if(i == solution[k] || //same column or same k=1/-1 (k=(y2-y1)/(x2-x1))
abs(i-solution[k])==abs((int)solution.size()-k)){
selected = false;
break;
}
}
if(selected)result.push_back(i);
}
return result;
}
void dfs(int n,int start,int &sum,vector<int>&solution){
if(start == n){ //find one solution,reach last line
sum++;
return;
}
vector<int>candiates = getNextCandiates(solution,n);
for(int k = 0;k < candiates.size();k++){
solution.push_back(candiates[k]);
dfs(n,start+1,sum,solution);
solution.pop_back();
}
}
int totalNQueens(int n) {
vector<int>solution;
int sum = 0;
dfs(n,0,sum,solution);
return sum;
}
};
相关文章推荐
- 书评:《算法之美( Algorithms to Live By )》
- 动易2006序列号破解算法公布
- Ruby实现的矩阵连乘算法
- C#插入法排序算法实例分析
- 超大数据量存储常用数据库分表分库算法总结
- C#数据结构与算法揭秘二
- C#冒泡法排序算法实例分析
- 算法练习之从String.indexOf的模拟实现开始
- C#算法之关于大牛生小牛的问题
- C#实现的算24点游戏算法实例分析
- c语言实现的带通配符匹配算法
- 浅析STL中的常用算法
- 算法之排列算法与组合算法详解
- C++实现一维向量旋转算法
- Ruby实现的合并排序算法
- C#折半插入排序算法实现方法
- 基于C++实现的各种内部排序算法汇总
- C++线性时间的排序算法分析
- C++实现汉诺塔算法经典实例
- PHP实现克鲁斯卡尔算法实例解析