N皇后问题

Description

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

Input

一个整数N，表示要把N个皇后摆放在一个N行N列的国际象棋棋盘上

Output

所有的摆放放案。每个方案一行，依次是第1行皇后位置、第2行皇后位置…第N行皇后位置。
多种方案输出顺序如下：优先输出第1行皇后列号小的方案。如果两个方案第1行皇后列号一致，那么优先输出第1行皇后列号小的方案…以此类推

Sample Input

1
4

Sample Output

2 4 1 3
3 1 4 2

算法题解

• 见代码注释

#include<bits/stdc++.h>

define num 100;

using namespace std;

int n; //当前棋盘的行数

/**
 * queen[i] i 表示第i行 queen[i]=j 表示第i的皇后放在第j列
*/
int queen[num];

void find_queen(int k){
    //k == n时表示前面0~k-1行共k行均已排好，此时只需要输出就行
    //从0行0列则不需要+1输出 
    if(k == n){
        for(int i = 0; i < n-1; i++){
            cout<<queen[i] + 1<<" ";
        }
        cout<<queen[n-1] + 1<<endl;
        return ;
    }
    //对于没排完的，找到一个当前排与前k-1排不冲突的位置
    int i, j;
    for(j = 0; j < n; j++){
        //不冲突条件：每两个皇后不在同一列、同一行或是某个正方形的对角上
        //这里肯定不会出现在同一行
        for(i = 0; i < k; i++){
            if(queen[i] == j || abs(i - k) == abs(queen[i] - j))
                break;
        }
        //当循环出来后未执行break说明当前行当前点与之前不冲突
        //再往下一行找呗
        if(i == k){
        	queen[k] = j;
            find_queen(k+1);
        }
    }
    return;
}

int main(){
    int N;
    cin>>N;
    while(N--){
        cin>>n;
        if(n < 4 && n != 1){
        	cout<<"NO ANSWER"<<endl;
        	continue;
        }
		find_queen(0);
    }
    return 0;
}