N皇后问题(dfs)

Problem:N皇后问题

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

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

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

Sample Input:

1
8
5
0

Sample Output:

1
92
10

Language:C++

#include <iostream>
#include <cstring>

using namespace std;

bool vis[3][21];
int ans;
int f[11];

void dfs(int row,int N)
{
  if(row==N) ans++;

  for(int column=1;column<=N;column++)
  {
    if(!vis[0][column]&&!vis[1][row+column]&&!vis[2][row+(N-column)])//分别对应列、主对角线、反对角线
    {
      vis[0][column]=vis[1][row+column]=vis[2][row+(N-column)]=1;
      dfs(row+1,N);
      vis[0][column]=vis[1][row+column]=vis[2][row+(N-column)]=0;
    }
  }
}

int main()
{
  for(int i=1;i<=10;i++)//预先将算出的方案数保存在f数组中，否则会tle
  {
    memset(vis,0,sizeof(vis));
    ans=0;
    dfs(0,i);
    f[i]=ans;
  }

  int N;
  while(cin>>N&&N)  cout<<f[N]<<endl;

  return 0;
}