DFS-回溯与剪枝-C - N皇后问题

    技术2022-07-10  155

    C - N皇后问题

    在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。 你的任务是,对于给定的N,求出有多少种合法的放置方法。  

    Input共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。Output共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。Sample Input

    1 8 5 0

    Sample Output

    1 92 10 #include <iostream> #include <algorithm> #include <cmath> using namespace std; int queen[12], q[12]={0}, N, sum=0; int check(int n) { for(int i=0; i<n; i++) if(queen[i]==queen[n] || abs(queen[i]-queen[n])==abs(n-i)) return 0; return 1; } void DFS(int n) { for(int i=0; i<N; i++){ queen[n] = i; if(check(n)){ if(n == N-1) sum++; else DFS(n+1); } } } int main(){ while(scanf("%d",&N)!=EOF){ if(N==0) break; if(q[N]!=0) printf("%d\n",q[N]); else{ DFS(0); q[N] = sum; printf("%d\n",sum); sum = 0; } } return 0; }

    (1)打表。在main()中提前算出来从1到10的所有N皇后问题的答案,并存储在数组中,等读取输入后立刻输出。如果不打表,而是输入N后再单独计算输出,会超时。

    (2)递归搜索DFS()。递归程序十分简洁,把第一个皇后按行放到棋盘上,然后递归放置其他的皇后,直到放完。

    (3)回溯判断check()。判断新放置的皇后和已经放好的皇后在横向、纵向、斜对角方向是否冲突。其中,横向并不需要判断,因为在递归的时候已经是按不同行放置的。

    (4)模块化编程。例如check()的内容很少,其实可以直接写在DFS()内部,不用单独写成一个函数。但是单独写成函数,把功能模块化,好处很多,例如逻辑清晰,容易差错等。建议在写程序的时候尽量把能分开的功能单独写成函数,这样可以大大减少编码和调试的时间。

     

    Processed: 0.040, SQL: 9