【dfs+回溯】hdu 2553 N皇后问题

Source :hdu 2553 N皇后问题

http://acm.hdu.edu.cn/showproblem.php?pid=2553

Problem Description
在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。
你的任务是，对于给定的N，求出有多少种合法的放置方法。

Input
共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

Output
共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

题意

N皇后问题，求解可行解的个数。

示例

Sample Input

1
8
5
0

Sample Output

1
92
10

---

思路

典型的回溯算法
dfs1-N行，判断是否同时满足不同列、不同主对角线和副对角线，满足条件的计数，开3个标记数组，vis[ 0 ][ i ]判断 第 i 列是否被占、vis[ 1 ][ cur - i + 8 ]判断主对角线、vis[ 2 ][ cur + i ]判断副对角线。

还有，这道题要提前打表，测试数据有重复！

主对角线有 （行 - 列 ）相等

8	7	6	5	4	3	2	1
9	8	7	6	5	4	3	2
10	9	8	7	6	5	4	3
11	10	9	8	7	6	5	4
12	11	10	9	8	7	6	5
13	12	11	10	9	8	7	6
14	13	12	11	10	9	8	7
15	14	13	12	11	10	9	8

副对角线有 （行 + 列 ）相等

2	3	4	5	6	7	8	9
3	4	5	6	7	8	9	10
4	5	6	7	8	9	10	11
5	6	7	8	9	10	11	12
6	7	8	9	10	11	12	13
7	8	9	10	11	12	13	14
8	9	10	11	12	13	14	15
9	10	11	12	13	14	15	16

参考小白书P123

---

参考代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<vector>
#include<map>
#include<climits>
#define clr(k,v) memset(k,v,sizeof(k))
#define ll long long
#define eps 1e-8
using namespace  std;

int n,ans,res[15];
bool vis[3][20];

void dfs(int cur){//cur表示递归深度，或理解为第几行
  if(cur == n + 1){//前n行已填满
    ans++;
    return;
  }
  for(int i = 1; i <= n; ++ i)//当前cur行，从第一个开始填
  if(!vis[0][i]&&!vis[1][cur-i+n]&&!vis[2][cur+i]){
    vis[0][i] = vis[1][cur-i+n] = vis[2][cur+i] = 1;
    dfs(cur+1);
    vis[0][i] = vis[1][cur-i+n] = vis[2][cur+i] = 0;//回溯
  }
}

void init(){//初始化打表
  for(int i = 1; i <= 10; ++ i){
    clr(vis,0);
    ans = 0;
    n = i;
    dfs(1);
    res[i] = ans;
  }
}

int main(){
  #ifndef ONLINE_JUDGE
  freopen("input.txt","r",stdin);
  #endif // ONLINE_JUDGE
  init();
  while(scanf("%d",&n) == 1 && n){
    printf("%d\n",res[n]);
  }
  return 0;
}

• 加粗 Ctrl + B
• 斜体 Ctrl + I
• 引用 Ctrl + Q
• 插入链接 Ctrl + L
• 插入代码 Ctrl + K
• 插入图片 Ctrl + G
• 提升标题 Ctrl + H
• 有序列表 Ctrl + O
• 无序列表 Ctrl + U
• 横线 Ctrl + R
• 撤销 Ctrl + Z
• 重做 Ctrl + Y