hdu2553 n皇后问题 dfs搜索 记忆化

N皇后问题

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 21710    Accepted Submission(s): 9709

Problem Description

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

 

Input

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

 

Output

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

 

Sample Input

  
  

   
   1
8
5
0
  
  

 

Sample Output

  
  

   
   1
92
10
  
  

 

Author

cgf

 

Source

2008 HZNU Programming Contest

 

不难，测试数据里重复的很多，因此要打个表记录一下结果

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
const int maxn = 20;
const int inf = 0x3f3f3f3f;
bool vis[maxn][maxn];
int n,m;
int times[maxn];
int res;
bool check(int x,int y){
    int i;
    for(i=1;i<=n;i++){
        if(!vis[i][y]&&i!=x){
            return false;
        }
    }
    for(i=1;;i++){
        if(x-i==0||y-i==0){
            break;
        }
        if(!vis[x-i][y-i]){
            return false;
        }
    }
    for(i=1;;i++){
        if(x+i>n||y+i>n){
            break;
        }
        if(!vis[x+i][y+i]){
            return false;
        }
    }
    for(i=1;;i++){
        if(x+i>n||y-i==0){
            break;
        }
        if(!vis[x+i][y-i]){
            break;
        }
    }
    for(i=1;;i++){
        if(x-i==0||y+i>n){
            break;
        }
        if(!vis[x-i][y+i]){
            return false;
        }
    }
    return true;
}
void dfs(int x){
    int i;
    if(x>n){
        res++;
    }
    for(i=1;i<=n;i++){
        if(check(x,i)){
            vis[x][i] = false;
            dfs(x+1);
            vis[x][i] = true;
        }
    }
}
int main(){
    memset(times, inf, sizeof(times));
    while(~scanf("%d",&n)&&n){
        if(times[n]!=inf){
            printf("%d\n",times[n]);
            continue;
        }
        memset(vis, true, sizeof(vis));
        res = 0;
        dfs(1);
        times[n] = res;
        printf("%d\n",res);
    }
    
    return 0;
}