《算法笔记》n皇后问题

问题:n*n的棋盘上放置n个皇后,使得n个皇后不能互相攻击(不能在同一行、同一列、同一对角线上)

思路(暴力):枚举1-n的全排列,判断每个全排列是否合法

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int n,count1=0;
int p[110];
bool hashTable[110]={false};
void generatep(int index)
{
    if(index==n+1)
    {
        bool flag=true;//排列是否合法的标志
        for(int i=1;i<=n;i++)//枚举任意两个皇后
        {
            for(int j=i+1;j<=n;j++)
                {
                    if(abs(i-j)==abs(p[i]-p[j]))//约束条件,不在同一条对角线上
                        flag=false;
                }
        }
        if(flag) count1++;
        return;
    }

    for(int x=1;x<=n;x++)
    {
        if(hashTable[x]==false)
        {
            p[index]=x;
            hashTable[x]=true;
            generatep(index+1);
            hashTable[x]=false;
        }
    }
}
int main()
{
    scanf("%d",&n);
    generatep(1);
    printf("%d",count1);
    return 0;
}

 优化(回溯法):当放置一部分皇后时,已经不符合,就可以不用往下递归

 

void generatep(int index)
{
    if(index==n+1)//如果到达递归边界,就一定满足条件
    {
        count1++;
        return;
    }

    for(int x=1;x<=n;x++)
    {
        bool flag=true;
        if(hashTable[x]==false)
        {
            for(int pre=1;pre<index;pre++)//先判断x与之前的皇后有无冲突
            {
                if(abs(index-pre)==abs(x-p[pre]))
                {
                    flag=false;
                    break;
                }
            }
            if(flag)//如果没有冲突,就把x放入
            {
                p[index]=x;
                hashTable[x]=true;
                generatep(index+1);
                hashTable[x]=false;
            }
        }
    }
}

 

posted @ 2021-02-02 12:06  inss!w!  阅读(125)  评论(0编辑  收藏  举报