5976 Problem D 【递归入门】n皇后 问题（原始的8皇后问题）

                         问题 D: 【递归入门】n皇后 问题（原始的8皇后问题）

                                                                         [命题人 : 外部导入]

                                                       时间限制 : 1.000 sec  内存限制 : 128 MB

                                                                      解决: 506提交: 506统计

题目描述

       会下国际象棋的人都很清楚：皇后可以在横、竖、斜线上不限步数地吃掉其他棋子。如何将8个皇后放在棋盘上（有8 * 8个方格），使它们谁也不能被吃掉！这就是著名的八皇后问题。

 

输入

一个整数n（ 1 < = n < = 10 ) 

输出

每行输出对应一种方案，按字典序输出所有方案。每种方案顺序输出皇后所在的列号，相邻两数之间用空格隔开。如果一组可行方案都没有，输出“no solute!”

样例输入 Copy

4

样例输出 Copy

2 4 1 3
3 1 4 2

一、AC代码

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;

int row[11];
int column[11];
int visited[11][11];
int ans[11];
int n;
bool no;

void Print()
{
    for(int i=1; i<n; i++)
        printf("%d ",ans[i]);
    printf("%d\n", ans[n]);
}

void init_vi()
{
    for( int i=0; i<=n; i++ )
    {
        row[i] = 0;
        column[i] = 0;
        for( int j=0; i<=n; i++ )
            visited[i][j] = 0;
    }
}

void dfs(int index, int r)  //index用来标记答案存放在ans数组中的下标，同时用来判断什么时候打印。
                                      //r表示当前在棋盘中的“行”
{
    bool juge;     //juge用来判断斜方有没有皇后，有则为false
    if( index==n+1 )
    {
        Print();
        no = false;
        return;
    }

        //每一次递归都从第一列开始
        for( int j=1; j<=n; j++)
        {
            juge = true;
            if( !row[r] && !column[j] && !visited[r][j] )  //row数组表示行是否已访问，column表示列是否已访问，visited是点标记数组
            {
                for( int k=r-1, w=j-1; k>0&&w>0; k--, w-- )  //判断左斜上方是否有皇后
                    if( visited[k][w] )
                    {
                        juge = false;
                        break;
                    }

                for( int k=r-1, w=j+1; k>0&&w<=n; k--, w++ )  //判断右斜上方是否有皇后
                    if( visited[k][w] )
                    {
                        juge = false;
                        break;
                    }

                if(juge)  //此棋盘点可以放置皇后
                {
                    row[r] = 1;
                    column[j] = 1;
                    visited[r][j] = 1;
                    ans[index] = j;
                    dfs(index+1, r+1);

                    visited[r][j] = 0;  //探寻完后回溯
                    row[r] = 0;
                    column[j] = 0;
                }
            }
        }
}

int main()
{
    while(cin>>n)
    {
        no = true;  //bool值no为true时代表没有解决方案
        init_vi();
        dfs(1,1);
        if(no)
            printf("no solute!\n");
    }
}

二、思路

    一开始一直在想如何同时下棋又同时判断。如果先暴力下棋再去dfs判断的话实在是太浪费时间。而判断同一行同一列是否有其他皇后实际上很简单，用一个数组的下标代表那一行/列就能解决，困难在于如何解决一边下棋一边判断斜方是否有其他八皇后的问题。后来思考了一阵，发现如果要实现起来其实也很简单。一开始原本就是从上到下去安排皇后，那么我们只需判断那些下过的皇后中有没有在该点的斜方就可以了 ----- 也就是判断某一点的左斜方和右斜方是否有其他皇后即可。

三、总结

    1. 每个解题思路要理解就看是否能懂开头和结尾的输入输出的逻辑。而一般题目的输入和输出如果多样的话都会有按字典序输出的要求，其实这为我们做题提供了便利（一些排序困难时间复杂度高的算法不会浪费时间去动手）