算法训练 王、后传说

版权声明：菜鸟一枚~~ 有想法可在下面评论， 转载标明出处即可。 https://blog.csdn.net/KLFTESPACE/article/details/82500960

问题描述

　　地球人都知道，在国际象棋中，后如同太阳，光芒四射，威风八面，它能控制横、坚、斜线位置。
　　看过清宫戏的中国人都知道，后宫乃步步惊心的险恶之地。各皇后都有自己的势力范围，但也总能找到相安无事的办法。
　　所有中国人都知道，皇权神圣，伴君如伴虎，触龙颜者死......
　　现在有一个n*n的皇宫，国王占据他所在位置及周围的共9个格子，这些格子皇后不能使用（如果国王在王宫的边上，占用的格子可能不到9个）。当然，皇后也不会攻击国王。
　　现在知道了国王的位置（x,y）（国王位于第x行第y列，x,y的起始行和列为1），请问，有多少种方案放置n个皇后，使她们不能互相攻击。

输入格式

　　一行，三个整数，皇宫的规模及表示国王的位置

输出格式

　　一个整数，表示放置n个皇后的方案数

样例输入

8 2 2

样例输出

10

数据规模和约定

　　n<=12

#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;
int n;
int a[14][14] = { 0 };
int t = 1, cnt = 0;
void print()
{
    printf("No. %d\n", t++);
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            printf("%d ", a[j][i]);
        }
        printf("\n");
    }
}
bool judge(int x, int y)
{
    for(int i=0; i<x; i++)
    {//竖直方向||左上方 || 右上方
        if(a[i][y]==1 || (y-i-1>=0 && a[x-i-1][y-i-1]==1) || (y+i+1<n && a[x-i-1][y+i+1]==1))
            return false;
     }
    return true;
}
/*
void dfs(int num)
{
    if (num >= 8){
        print();
    }
    for (int i = 0; i < 8; i++)
    {
       if (a[num][i]!=1&&judge(num, i))
        {
            a[num][i] = 1;
            dfs(num + 1);
            a[num][i] = 0;
        }
    }
}
*/

int dfs(int num)
{
    if (num >= n){
      //  print();
        cnt++;
   //     return 0;
    }
    for (int i = 0; i < n; i++)
    {
        if (a[num][i]==0&&judge(num, i))
        {
            a[num][i] = 1;
            if(dfs(num + 1))
                return 1;
            a[num][i] = 0;
        }

    }
return 0;
}
int main()
{
int x, y;
    cin >> n;
    cin >> x >> y;
    x -= 1;
    y -= 1;
    a[x][y] = 2;
    a[x-1][y] = 2;
    a[x+1][y] = 2;
    a[x-1][y+1] = 2;
    a[x][y-1] = 2;
    a[x-1][y-1] = 2;
    a[x][y+1] = 2;
    a[x+1][y+1] = 2;
    a[x+1][y-1] = 2;


    dfs(0);
    cout << cnt << endl;
    return 0;
}