n皇后(回溯法)C/C++

时间限制: 1 Sec 内存限制: 128 MB

题目描述
在n*n的棋盘摆放n个棋子，任意两个棋子都不能处于同一行、同一列或同一斜线上，问有多少种摆法。

输入
输入一个整数n（1<=n<=8），表示棋子的个数。

输出
输出一个整数，表示棋子的摆法数目。

样例输入
4
样例输出
2

AC代码：

#include<bits/stdc++.h>
using namespace std;
int n,cnt=0,a[20][20];
bool used[20];
bool judge(int x,int y){
   
    if(used[y])return false;
    
    return true;
}
bool solve(int x,int y){
    //cout<<x<<" "<<y<<" "<<endl;;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            //cout<<a[i][j]<<" ";
            if(a[i][j]&&(abs(x-i)==abs(y-j))){
                    //cout<<endl;
                return false;
            }
        }
        //cout<<endl;
    }
    //cout<<endl;
    return true;
}
void dfs(int x,int ans){
    if(ans==n){
        cnt++;
        return ;
    }
    if(x==n+1)return ;
    for(int i=1;i<=n;i++){
        if(judge(x,i)&&solve(x,i)){
           //cout<<judge(x,i)<<endl;
            used[i]=true;
            a[x][i]=1;
            dfs(x+1,ans+1);
            a[x][i]=0;
            used[i]=false;
        }
    }
    //dfs(x+1,ans);//若这行没放
}
int main()
{
    cin>>n;
    dfs(1,0);
   /*for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cout<<a[i][j]<<" ";
        }
        cout<<endl;
    }
    for(int i=1;i<=n;i++){
        cout<<used[i]<<" ";
    }
    cout<<endl;*/
    cout<<cnt<<endl;
    return 0;
}