HDU 2553 N皇后问题 深搜

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

题意：

　　给出个N，代表N*N表格，在里面放N个皇后，要保证每个皇后之间不会攻击到，皇后的攻击范围：自己所在的那一行，自己所在的那一列，自己所在的两条

45度对角线。

 

坑爹：

　　如果输入一个N就去搜一次的话，数据给出每一次的N都是10的话就会超时。

 

解法：

　　反正这题N最大才10，就把1到10的结果先保存在数组中，然后每输入一次N就利用下表访问一次数组得到答案。

 

#include<iostream>
using namespace std;

int N;
int count = 0;
int num[20];
int A[20] = {0};

void DFS(int mid)
{
    if(mid == N )
    {
        count++;
    }
    else
    {
        for(int i = 0 ; i < N ; i ++ )
        {
            int flag = 1;
            A[mid] = i;
            for(int j = 0 ; j < mid ; j ++ )
            {
                if(A[mid] == A[j] ||  mid - A[mid] == j - A[j] || mid + A[mid] == j + A[j] )
                {
                    flag = 0;
                    break;
                }
            }
            if(flag)
            {
                DFS(mid + 1);
            }
        }
    }
}

int main()
{
    for(int i = 0 ; i < 10 ; i ++ )
    {
        count = 0;
        memset(A,0,sizeof(A));
        N = i + 1;
        DFS(0);
        num[i] = count;
    }
    while(cin>>N,N)
    {
        cout<<num[N - 1]<<endl;
    }
    
    return 0;
}