hdu 2553

N皇后问题

http://acm.hdu.edu.cn/showproblem.php?pid=2553

经典深搜：在棋盘上放置棋子，行、列、斜45度不重复

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>

using namespace std;
int count[15];//n的最大范围是10，打表！
int cal;//k个大小的棋盘放的数目
int map[15][15];//棋盘

bool dfs(int row,int column,int k)
{
    //最后一行也合适，放置数目加1
    if(row>k)
    {
        cal++;
        return 1;
    }
    //判断同一列是否有棋子
    for(int i=1;i<row;i++)
        if(map[i][column])
            return 0;
    //判断左上45度是否有棋子
    for(int i=row-1,j=column-1;i>0&&j>0;i--,j--)
    {
        if(map[i][j])
            return 0;
    }
    //判断右上45度是否有棋子
    for(int i=row-1,j=column+1;i>0&&j<=k;i--,j++)
    {
        if(map[i][j])
            return 0;
    }
    //都通过，该点合适并判断下一行
    map[row][column]=1;
    for(int i=1;i<=k;i++)
    {
        //当dfs(row+1,i,k)为1时，改行为最后一行，棋盘已放满
        if(dfs(row+1,i,k))
            break;
    }
    //该点判断完成，恢复后再去判断其他点
    map[row][column]=0;
    return 0;
}
int main()
{
    memset(map,0,sizeof(map));
    for(int k=1;k<=10;k++)
    {
        cal=0;
        count[k]=0;
        for(int i=1;i<=k;i++)
            dfs(1,i,k);
        count[k]=cal;
    }
    //for(int i=1;i<=10;i++)
        //printf("%d %d\n",i,count[i]);
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)
            break;
        printf("%d\n",count[n]);
    }
    return 0;
}