P1219 八皇后 回溯法题解和思路

题目描述：

检查一个如下的6 x 6的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。

上面的布局可以用序列2 4 6 1 3 5来描述，第i个数字表示在第i行的相应位置有一个棋子，如下：

行号 1 2 3 4 5 6

列号 2 4 6 1 3 5

这只是跳棋放置的一个解。请编一个程序找出所有跳棋放置的解。并把它们以上面的序列方法输出。解按字典顺序排列。请输出前3个解。最后一行是解的总个数。

 

题解代码：

#include <iostream>

using namespace std;

int hang[100];
int b[100];
int c[100];
int d[100];

int N;
int ans = 0;

void Search(int);

int main(void)
{
    cin >> N;
    Search(1);
    cout << ans << endl;
    return 0;
}

void Search(int cur)
{
    if(cur > N)
    {
        if(ans < 3)
        {
            for(int i = 1;i <= N;i++)
            {
                cout << hang[i] << ' ';
            }
            cout << endl;
        }
        ans++;
        return;
    }
    else
    {
    
        for(int i = 1; i <= N;i++)
        {
            if(!(b[i]) && (!c[i+cur]) && (!d[i-cur+N]))
            {
                hang[cur] = i;	//第cur行的i列被占领 
                b[i] = 1;	//第i列被占领 
                c[i+cur] = 1;	//某对角线占领，只是某种表示法 
                d[i-cur+N] = 1;	//同上 
                Search(cur+1);
                b[i] = 0;	
                c[i+cur] = 0;
                d[i-cur+N] = 0;
                //回溯 
            }
        }
    }
}

思路：用一个数组hang表示某一行的皇后占据在第i列.c，d两个数组表示对角线被占据，在下一次的DFS过程中不能放在特定的位置，b数组表示第i列被占据。

用深度优先搜索求出结果，ans表示解的数目，当ans< 3时，打印解.