【codevs1295 N皇后问题（不输出方案）】回溯法

最新推荐文章于 2020-10-03 21:30:32 发布

原创最新推荐文章于 2020-10-03 21:30:32 发布 · 176 阅读

0 ·

CC 4.0 BY-SA版权

题目专栏收录该内容

81 篇文章

订阅专栏

本文提供了两种不同的算法实现来解决N皇后问题：一种使用一维数组进行存储和判断，另一种采用二维数组的方法，并强调了数组大小的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

int n, C[15], tot = 0;

void search(int cur) {
    if (cur == n) tot++;
    else for (int i = 0; i < n; i++)
    {
        int ok = 1;
        C[cur] = i; //尝试把cur行的皇后放在第i列
        for (int j = 0; j < cur; j++)
        {
            if (C[cur] == C[j] || cur - C[cur] == j - C[j] || cur + C[cur] == j + C[j])
            {
                ok = 0;
                break;
            }
        }
        if (ok) search(cur + 1);
    }
}

int main()
{
    scanf("%d", &n);

    search(0);

    printf("%d\n", tot);

    return 0;
}

第一种，用一维数组存储判断。

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

int vis[3][30], n, tot = 0, C[30];

void search (int cur)
{
    if (cur == n)
    {
        tot++;
        // for (int i = 0; i < n; i++)
        // {
        //     printf("%d ", C[i]);
        // }
        // printf("\n");
    }
    else for (int i = 0 ; i < n; i++)
    {
        if (!vis[0][i] && !vis[1][cur + i] && !vis[2][cur - i + n]) //利用二维数组直接判断
        {
            C[cur] = i;
            vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 1;
            search(cur + 1);
            vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 0;
        }
    }
}

int main()
{
    //memset(vis, 0, sizeof(vis));

    scanf("%d", &n);

    search(0);

    printf("%d\n", tot);

    return 0;
}