基础训练 2n皇后问题

2n皇后问题

#include<iostream>
#include<vector>
using namespace std;
int cnt = 0, n;
vector< vector<int> > vi(9, vector<int>(9,0) ); //储存棋盘
vector<int> row1(9, 0), row2(9, 0), md1(19, 0), md2(19, 0), vd1(19, 0), vd2(19, 0); // row1记录每行是否有放了黑棋， row2记录每行是否有放了白棋。 md1记录每条主对角是否有放了黑棋，md2记录每条主对角线是否有放了白棋。vd1记录每条副对角线是否有放了黑棋，vd2记录每条副对角线是否有放了白棋。
void DFS(int deep){   //回溯
    if(deep == n+1){  //成功摆放完，cnt++
        cnt++;
        return ;
    }
    for(int i=1; i<=n; i++){ 
        if(vi[deep][i] && !row1[i] && !md1[deep-i+n] && !vd1[deep+i]){ //判断此处是否可以摆放棋子，且所在行，对角线处是否有同颜色的棋子，列不用，因为每列就放该颜色一枚棋子
            vi[deep][i]=0; row1[i]=1; md1[deep-i+n]=1; vd1[deep+i]=1; //修改数据
                for(int j=1; j<=n; j++){ //处理另一种颜色的棋子
                    if(vi[deep][j] && !row2[j] && !md2[deep-j+n] && !vd2[deep+j]){
                        vi[deep][j]=0; row2[j]=1; md2[deep-j+n]=1; vd2[deep+j]=1;
                            DFS(deep+1);
                        vi[deep][j]=1; row2[j]=0; md2[deep-j+n]=0; vd2[deep+j]=0; //改回原来的数据，防止数据变脏
                    }
                }
            vi[deep][i]=1; row1[i]=0; md1[deep-i+n]=0; vd1[deep+i]=0; 
        }
    }
}
int main(){
    cin>>n;
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            cin>>vi[i][j];
    DFS(1);
    cout<<cnt<<endl;
    return 0; 
}

转载于:https://www.cnblogs.com/A-Little-Nut/p/10341578.html