AcWing 843. n-皇后问题

原创已于 2024-01-14 13:14:19 修改 · 409 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #深度优先 #图论

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

在这里插入图片描述

#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;
}