1019 N皇后问题

1019 N皇后问题

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

思路：考虑皇后放置的位置，对于每一行，需要枚举每个可以放置皇后的位置，而且需要判断当前位置（第i行）是否满足条件，即判断这个位置是否与放置好的前i-1行的皇后的位置相冲突，如果冲突，说明这个位置不合适；否则的话，就可以枚举下一行皇后的位置，直至第n行。

感想：放置皇后与之前位置的判断，使其不产生冲突，不断的回溯求解。

#include<iostream>

#include<math.h>

using namespace std;

int a[11],n,sum,flag;

void dfs(int k){

    if(k>n){

        sum++;

        return ;

    }

    else{

        for(int i=1;i<=n;i++){

            a[k]=i;

            flag=1;

            for(int j=1;j<k;j++){

                if(a[j]==a[k]||abs(k-j)==abs(a[k]-a[j])){

                    flag=0;

                    break;

                }

            }

            if(flag)

                dfs(k+1);

        }

    }

}

int main(){

   int cnt[11];

   for(n=1;n<11;n++){

    sum=0;

    dfs(1);

    cnt[n]=sum;

   }

   while(cin>>n&&n){

    cout<<cnt[n]<<endl;

   }

   return 0;

}