题目:

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

代码:

class Solution {
public:
int totalNQueens(int n)
{
int ret = ;
if ( n== ) return ret;
vector<bool> colUsed(n,false), diagUsed1(*n-,false), diagUsed2(*n-,false);
Solution::dfs(ret, , n, colUsed, diagUsed1, diagUsed2);
return ret;
}
static void dfs( int &ret, int row, int n, vector<bool>& colUsed, vector<bool>& diagUsed1, vector<bool>& diagUsed2 )
{
if ( row==n ) { ret++; return; }
for ( size_t col = ; col<n; ++col ){
if ( !colUsed[col] && !diagUsed1[col+n--row] && !diagUsed2[col+row] )
{
colUsed[col] = diagUsed1[col+n--row] = diagUsed2[col+row] = true;
Solution::dfs(ret, row+, n, colUsed, diagUsed1, diagUsed2);
diagUsed2[col+row] = diagUsed1[col+n--row] = colUsed[col] = false;
}
}
}
};

tips:

如果还是用深搜的思路,这个N-Queens II要比N-Queens简单一些,可能是存在其他的高效解法。

===========================================

第二次过这道题,经过了N-Queens,这道题顺着dfs的思路就写下来了。

class Solution {
public:
int totalNQueens(int n)
{
int ret = ;
if ( n< ) return ret;
vector<bool> colUsed(n, false);
vector<bool> r2l(*n-, false);
vector<bool> l2r(*n-, false);
Solution::dfs(ret, n, , colUsed, r2l, l2r);
return ret;
}
static void dfs(
int& ret,
int n,
int index,
vector<bool>& colUsed,
vector<bool>& r2l,
vector<bool>& l2r)
{
if ( index==n )
{
ret++;
return;
}
for ( int i=; i<n; ++i )
{
if ( colUsed[i] || r2l[i-index+n-] || l2r[i+index] ) continue;
colUsed[i] = true;
r2l[i-index+n-] = true;
l2r[i+index] = true;
Solution::dfs(ret, n, index+, colUsed, r2l, l2r);
colUsed[i] = false;
r2l[i-index+n-] = false;
l2r[i+index] = false;
}
}
};

最新文章

  1. EF里查看/修改实体的当前值、原始值和数据库值以及重写SaveChanges方法记录实体状态
  2. tomcat端口号被占用或者修改端口号的解决方法
  3. cocos2d-x 内存管理浅析
  4. android瀑布流效果(仿蘑菇街)
  5. Coding 初级教程(二)——上传已有项目
  6. CISCO动态VLAN配置
  7. 第三百零四天 how can I 坚持
  8. linux mysql 安装(rpm)
  9. dm3730和dm6437,dm6446,AM335x启动过程的不同
  10. mytest 截图
  11. 我做的第一个程序(菜鸟的java课堂笔记)
  12. C++ 虚指针、成员变量与类对象的偏移地址
  13. 一文入门NodeJS
  14. Java_String&amp;StringBuilder&amp;StringBuffer类
  15. Linux系统网络文件配置
  16. permutohedral lattice理解
  17. react与umi
  18. POJ 2718【permutation】
  19. django rest framework restful 规范
  20. C#系统登录随机验证码生成及其调用方法

热门文章

  1. uLua学习之调用Lua函数(五)
  2. 关于配置httpd2.4.18+php5.6
  3. System Center Configuration Manager 2016 必要条件准备篇(Part1)
  4. python内存泄露的诊断(转)
  5. java的图形界面初学惯用
  6. 又一次摔MFC坑里了
  7. 进程加载与segment
  8. 漫谈 Clustering (番外篇): Expectation Maximization
  9. python 线程even
  10. pandas 代码