演算法思路:DFS。
這一題可以用DFS + 回溯的方法進行暴力枚舉。但還是要注意可行性的判斷,及時的break以減少時間開銷。
要注意的是雖然n最大隻有10,但是case會很多,如果每次都DFS會TLE,所以要打表。這個問題只能怪測試資料太無聊了。
自己剛開始做的時候沒有注意vis數組下標最大會達到 2 * n - 1,導致數組開得太小,訪問了非法記憶體,WA了,可怕的是像這種情況系統並不會有任何提示,照樣運行,還好及時發現,下次吸取教訓。
//模板開始#include <string> #include <vector> #include <algorithm> #include <iostream> #include <sstream> #include <fstream> #include <map> #include <set> #include <cstdio> #include <cmath> #include <cstdlib> #include <ctime>#include<iomanip>#include<string.h>#define SZ(x) (int(x.size()))using namespace std;int toInt(string s){istringstream sin(s); int t; sin>>t; return t;}template<class T> string toString(T x){ostringstream sout; sout<<x; return sout.str();}typedef long long int64;int64 toInt64(string s){istringstream sin(s); int64 t; sin>>t;return t;}template<class T> T gcd(T a, T b){ if(a<0) return gcd(-a, b);if(b<0) return gcd(a, -b);return (b == 0)? a : gcd(b, a % b);}//模板結束(通用部分)#define ifs cinint n;#define MAXN 25int C[MAXN];int tot;int vis[3][MAXN];int dabiao[MAXN];void search(int cur){int i, j;if(cur == n){tot++;}else{for(i = 0; i < n; i++){if(!vis[0][i] && !vis[1][cur + i] && !vis[2][cur - i + n]){C[cur] = i;vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 1;search(cur + 1);vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 0;}}}}//【練習07】 DFS 1024 N皇后問題int main(){//ifstream ifs("shuju.txt", ios::in);for(n = 1; n <= 12; n++)//打表。{tot = 0;memset(vis, 0, sizeof(vis));search(0);dabiao[n] = tot;}while(ifs>>n && n != 0){cout<<dabiao[n]<<endl;}return 0;}