N皇后问题递归回溯



N皇后问题

Time Limit : 2000/1000ms (Java/Other)   Memory Limit : 32768/32768K (Java/Other)

Total Submission(s) : 20   Accepted Submission(s) : 9

Font: Times New Roman | Verdana | Georgia

Font Size: ← →

Problem Description

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

Input

共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

Output

共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

Sample Input

1
8
5
0

Sample Output

1
92
10

#include <iostream>
#include<cstdio>
using namespace std;
int n,ant,c[10];
void search(int r)
{
    if(r==n)
    {
        ant++;
        return ;
    }
    else
    {
        for(int i=0; i<n; i++)
        {
            c[r]=i;///当前列取遍0,1,2,3,4,5,6,7
            int flag=1;
            for(int j=0; j<r; j++) ///当前行是r行，列是j列
            {
                if(c[r]==c[j]||r-j==c[r]-c[j]||r-j==c[j]-c[r])///如果当前列与之前有冲突
                {
                    flag=0;
                    break;
                }
            }
            if(flag)///无冲突则继续放下一行
                search(r+1);
        }
    }
}
int main()
{
    while(scanf("%d",&n)!=EOF&&n)
    {
        ant=0;
        search(0);
        cout<<ant<<endl;
    }
    return 0;
}

若要打印出N皇后图形，再添加print()函数即可：

#include <iostream>
#include<cstdio>
using namespace std;
int n,ant,c[10];
void print()
{
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            if(j==c[i])
                printf("1 ");
            else
                printf("0 ");
        }
        printf("\n");
    }
    printf("\n");
}
void search(int r)
{
    if(r==n)
    {
        print();
        ant++;
        return ;
    }
    else
    {
        for(int i=0; i<n; i++)
        {
            c[r]=i;///当前列取遍0,1,2,3,4,5,6,7
            int flag=1;
            for(int j=0; j<r; j++) ///当前行是r行，列是j列
            {
                if(c[r]==c[j]||r-j==c[r]-c[j]||r-j==c[j]-c[r])///如果当前列与之前有冲突
                {
                    flag=0;
                    break;
                }
            }
            if(flag)///无冲突则继续放下一行
                search(r+1);
        }
    }
}
int main()
{
    while(scanf("%d",&n)!=EOF&&n)
    {
        ant=0;
        search(0);
        cout<<ant<<endl;
    }
    return 0;
}