HDU 2553 N皇后问题（DFS）

题意：N*N的位置放N个皇后，每两个皇后不能在同一行同一列同一对角线，求摆放方法。
很明显每一行或者每一列只能放一个，因此可以从第一行开始，不断在每一行中求每一列是否能放；
伪代码：

void dfs(int row)
{
    if(n==row) //如果找到结果
    {
        //做相应操作
    }
    else{
        for(k=0 to n)//试探row行的每一列
        {
            if(can_place(row,k))
            {
                place(row,k);//放置皇后
                dfs(row+1); //继续探测下一行
            }
            //对已改变的值做相应的返回处理
        }
    }

}

接下来将伪代码实现：
因为我们求的是方案数，所以当有结果时ans++（1~n输入就用n+1作为终止条件，0~n-1输入就用n作为终止条件）
每一个点在一种情况访问完后恢复未访问的状态，vis[i][j]=0
中间判断是否放置就判断现在位置的所在行列对角线有没有其它皇后即可
然后这个方法不太优化。。所以直接会TLE，所以我把结果打表输出了……

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int vis[20][20];
int ans=0,n;
bool can_place(int row,int k){
    for(int i=1;i<=n;i++){
        if(vis[row][i]) return false;
    }
    for(int i=1;i<=n;i++){
        if(vis[i][k]) return false;
    }

    for(int i=-(n-1);i<=(n-1);i++){
        if((row+i>=1&&k+i>=1)){
            if(vis[row+i][k+i]) return false;           
        }
        if(row+i>n||k+i>n) break;
    }

    for(int i=(n-1);i>=-(n-1);i--){
        if(row+i>=1&&k-i>=1){
            if(vis[row+i][k-i]) return false;
        }
            if(row+i>n&&k-i>n) break;
    }
    return true;
}
void place(int row,int k){
    vis[row][k]=1;
}
void dfs(int row){
    if(row==(n+1)){
//      for(int i=1;i<=n;i++){
//          for(int j=1;j<=n;j++){
//              cout<<vis[i][j]<<" ";
//          }
//          cout<<endl;
//      }
        ans++;
    }
    else{
        for(int k=1;k<=n;k++){
            if(can_place(row,k)){
                place(row,k);
                dfs(row+1);
            }
            vis[row][k]=0;
        }

    }
}
int main(){
    int res[11]={0,1,0,0,2,10,4,40,92,352,724};
//  freopen("1.in","r",stdin);
//  freopen("1.out","w",stdout);
    while(~scanf("%d",&n)&&n){
//      memset(vis,0,sizeof(vis));
//      ans=0;
//      dfs(1);

        printf("%d\n",res[n]);
    }
}