n皇后问题

n皇后问题

　　八皇后问题是一个以国际象棋为背景的问题：如何能够在 8×8 的国际象棋棋盘上放置八个皇后，使得任何一个皇后都无法直接吃掉其他的皇后？为了达到此目的，任两个皇后都不能处于同一条横行、纵行或斜线上。八皇后问题可以推广为更一般的n皇后摆放问题：这时棋盘的大小变为n1×n1，而皇后个数也变成n2。而且仅当 n2 = 1 或 n1 ≥ 4 时问题有解。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define N 100
using namespace std;
int n, sum;
int check[N];
void res(int seq)
{
    int i, j, k;
    if(seq == n)
        sum++;
    else
        for(i = 0; i < n; i++){
            int k = 1;
            check[seq] = i;
            for(j = 0; j < seq; j++)
                if(check[seq] == check[j] || seq-check[seq] == j-check[j] || seq+check[seq] == j+check[j]){
                    k = 0;
                    break;
                }
            if(k)
                res(seq + 1);
        }
}
int main(void)
{
    int i, j, k;
    while(1){
        printf("请输入皇后数：(退出:-1)\n");
        scanf("%d", &n);
        if(n == -1)
            break;
        memset(check, 0, sizeof(check));
        sum = 0;
        res(0);
        printf("方案数：%d\n", sum);
    }
    return 0;
}

利用二维数组优化

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define N 100
using namespace std;
int n, sum;
int vis[N][N];
void res(int seq)
{
    int i, j, k;
    if(seq == n)
        sum++;
    else
        for(i = 0; i < n; i++){
            if(!vis[0][i] && !vis[1][seq+i] && !vis[2][seq-i+n]){
                    vis[0][i] = vis[1][seq+i] = vis[2][seq-i+n] = 1;
                    res(seq+1);
                    vis[0][i] = vis[1][seq+i] = vis[2][seq-i+n] = 0;
            }
        }
}
int main(void)
{
    int i, j, k;
    while(1){
        printf("请输入皇后数：(退出:-1)\n");
        scanf("%d", &n);
        if(n == -1)
            break;
        memset(vis, 0, sizeof(vis));
        sum = 0;
        res(0);
        printf("方案数：%d\n", sum);
    }
    return 0;
}