AcWing 843. n-皇后问题

honortech

已于 2024-01-14 13:14:19 修改

阅读量375

点赞数 9

文章标签：算法深度优先图论

于 2024-01-14 13:08:52 首次发布

本文链接：https://blog.youkuaiyun.com/honortech/article/details/135582160

版权

这篇文章详细介绍了如何使用C++编程语言中的深度优先搜索算法（DFS）来遍历和打印一个二维棋盘，通过占用标志控制路径。

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

在这里插入图片描述

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int N = 10;
int n;
int col[N], dg[N + N], rdg[N + N];      // 列占用
										// 正对角线占用
										// 副对角线占用
char table[N][N]; 						// 棋盘

void dfs(int index)
{
    if(index == n) // 结束条件
    {
        for (int i = 0; i < n; i ++ )
        {
            for (int j = 0; j < n; j ++ )
                cout << table[i][j];
            cout << endl;
        }
        cout << endl;    
        return;    
    }
    
    for (int j = 0; j < n; j ++ )
    {
        // n - index = b + j + 1    正对角线 (b会为负，要再加上n)
        // n - index = b - (j + 1)  副对角线 
        if(!col[j] && !dg[n - index - j - 1 + n] && !rdg[n - index + j + 1])
        {
            col[j] = dg[n - index - j - 1 + n] = rdg[n - index + j + 1] = 1;
            table[index][j] = 'Q';
            dfs(index + 1);
            table[index][j] = '.'; // 搜索完恢复
            col[j] = dg[n - index - j - 1 + n] = rdg[n - index + j + 1] = 0;
        }
    }
    
}
int main()
{
    cin >> n;
    for (int i = 0; i < n; i ++ )
        for (int j = 0; j < n; j ++ )
            table[i][j] = '.';
    
    dfs(0);
    
    return 0;
}