P1219 八皇后（DFS 回溯）

题目链接：P1219 八皇后

题解： 就是一道DFS搜索加回溯，关键就是判断横纵对角线有没有值；可以直接for循环循环地图判断，如代码1.0，但会超时；所以最好用2.0直接将对角线和横纵弄到一个数组，如2.0的图示；输出时可以判断前三个输出，并不一定要保存下来，如3.0所示；以前的题解：点击这里！

知识点： 对角线的判断可以 i + j 和 i - j + N 进行判断，如图所示。

参考代码1.0： 直接循环查找，最后一个点超时。

#include <string>
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;

//由于每次扫描检查次数太多，导致超时

int N, pos;
int s[20][20];
bool vis[20][20];
int res[100010][20];

bool in(int x, int y)
{
    return x >= 1 && x <= N && y >= 1 && y <= N;
}

//检查当前行和当前列有无占用以及所有对角线
int check(int a, int b)
{
    if(s[a][b]) return 1;
    //行和列
    for(int i = 1; i <= N; i++)
        if(s[a][i]) return 1;
    for(int i = 1; i <= N; i++)
        if(s[i][b]) return 1;
    //对角线
    int x = a, y = b;
    for(int i = 1; i <= N; i++)
    {
        x += 1, y += 1;
        if(!in(x, y)) break;
        if(s[x][y]) return 1;
    }
    x = a, y = b;
    for(int i = 1; i <= N; i++)
    {
        x -= 1, y -= 1;
        if(!in(x, y)) break;
        if(s[x][y]) return 1;
    }
    x = a, y = b;
    for(int i = 1; i <= N; i++)
    {
        x -= 1, y += 1;
        if(!in(x, y)) break;
        if(s[x][y]) return 1;
    }
    x = a, y = b;
    for(int i = 1; i <= N; i++)
    {
        x += 1, y -= 1;
        if(!in(x, y)) break;
        if(s[x][y]) return 1;
    }
    
    return 0;
}

void dfs(int a)
{
    if(a == N + 1)//N = 3
    {
        for(int i = 1; i <= N; i++)
            for(int j = 1; j <= N; j++)
                if(s[i][j]) {res[pos][i] = j; break;}
        pos ++; return;
    }
    //扫描当前行的列
    for(int i = 1; i <= N; i++)
    {
        if(!check(a, i))
        {
            s[a][i] = 1;
            dfs(a + 1);
            s[a][i] = 0;
        }
    }
}

void print()
{
    for(int i = 0; i < 3; i++)
    {
        for(int j = 1; j <= N; j++)
            cout << res[i][j] << " ";
        cout << endl;
    }
    cout << pos << endl;
}

int main()
{
    cin >> N;
    dfs(1);
    print();
    return 0;
}

参考代码2.0： 关于标记对角线的方法，i + j 和 i - j + N 。效果如图，即同一对角线数字都是一样的；同样，这样可以减少循环查找所需要的的时间。

#include <string>
#include <cstdio>
#include <ctime>
#include <iostream>
#include <algorithm>
using namespace std;

int N, pos;
int s[20][20], f[100], b[100], c[100], d[100];
bool vis[20][20];
int res[100010][20];

void dfs(int a)
{
    if(a == N + 1)//N = 3
    {
        for(int i = 1; i <= N; i++)
            for(int j = 1; j <= N; j++)
                if(s[i][j]) {res[pos][i] = j; break;}
        pos ++; return;
    }
    //扫描当前行的列
    for(int i = 1; i <= N; i++)
    {
        if(!b[i] && !c[a + i] && !d[a - i + N])
        {
            s[a][i] = 1;
            b[i] = 1; c[a + i] = 1;
            d[a - i + N] = 1;
            
            dfs(a + 1);
            
            s[a][i] = 0;
            b[i] = 0; c[a + i] = 0;
            d[a - i + N] = 0;
        }
    }
}

void print()
{
    for(int i = 0; i < 3; i++)
    {
        for(int j = 1; j <= N; j++)
            cout << res[i][j] << " ";
        cout << endl;
    }
    cout << pos << endl;
}

int main()
{
    cin >> N;
    dfs(1);
    print();
    return 0;
}

参考代码3.0： 直接输出前三个，无需保存

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;

int num=0,n;
int a[100],b[100],c[100],d[100];//行(存储行上皇后的下标) 列 主对角线 副对角线

void dfs(int i)//i为行 j为列
{
    if(i>n)//能走到这里就是一种情况
    {
        if(num<3)
        {
            for(int j=1;j<=n;j++)
            {
                cout<<a[j]<<" ";
            }
            cout<<endl;
        }
        num++;//记录解的个数
        return;
    }
    else
    {
        for(int j=1;j<=n;j++)//列数
        {
            if((!b[j])&&(!c[i+j])&&(!d[i-j+n]))
            {
                a[i]=j;//i为皇后数，也为行数，存储列下标
                b[j]=1;//标记此列已使用
                c[i+j]=1;//标记副对角线已使用
                d[i-j+n]=1;//标记主对角线已使用

                dfs(i+1);//找下一个皇后位置

                //执行到这里的条件：n个皇后都已经找到（一种情况）
                //或者没有找到下一个皇后return到此时的dfs

                //清除上一次找皇后做的标记
                b[j]=0;
                c[i+j]=0;
                d[i-j+n]=0;
                //回到for循环，找此轮的下一个可能位置
            }
        }
    }
}

int main()
{

    cin>>n;

    dfs(1);

    cout<<num<<endl;

    return 0;
}