回溯法（2）

在Ｎ行Ｎ列的数阵中, 数Ｋ（１〈＝Ｋ〈＝Ｎ）在每行和每列中出现且仅

出现一次，这样的数阵叫Ｎ阶拉丁方阵。例如下图就是一个五阶拉丁方阵。

编一程序，从键盘输入Ｎ值后，打印出所有不同的Ｎ阶拉丁方阵，并统计个数。

    1 2 3 4 5

    2 3 4 5 1

    3 4 5 1 2

    4 5 1 2 3

    5 1 2 3 4

 

这个算法和第一个差不多，每种可能都进行筛选，不过可能性更多，应该是N^(N*N)个吧。而且不能用上面那种笨办法了，循环N*N次，N都不能固定。只好按照回溯算法构造一个树出来，这个树应该是N*N层，每个父结点有N个子结点。遍历树的时候子结点是动态增加或删除的，hoho，要不然太占内存了也。

N=5的时候，实测了一下，总共有个161280拉丁方阵，在我的破T61上计算耗时7s，还过得去；N=6的时候就没耐心了. N=10的话，算了N久也没算完，总个数用64位好像都未必够，没兴趣了的说。

 

C++代码：

/* 1 2 3 4 5 2 3 4 5 1 3 4 5 1 2 4 5 1 2 3 5 1 2 3 4 */ //#include "header.h" #include <iostream> #include <time.h> using namespace std; #define LOG_ON 0 class array_4; class tree_4; static int max_size = 0; static array_4 *current = NULL; static __int64 total_cnt = 0; /* 二维数组，用于保存矩阵结果，附加了一些增加删除元素的功能 */ class array_4 { public: array_4(int val) : size(val), x(0), y(0), data(NULL) { data = new int*[val]; for(int i = 0; i < val; i++) data[i] = new int[val]; } ~array_4() { for(int i = 0; i < size; i++) delete [] data[i]; delete [] data; } public: bool push_back(int val) { for(int i = 0; i < x; i++) { if(data[i][y] == val) return false; } for(int i = 0; i < y; i++) { if(data[x][i] == val) return false; } if((x > size - 1) || (y > size - 1)) return false; data[x][y] = val; if(y < size - 1) { y++; } else { x++; y = 0; } return true; } void pop_back() { if((x < 0) || (y < 0)) return; if((x == 0) && (y == 0)) return; if(y > 0) { y--; } else { x--; y = size - 1; } } void show() { cout << endl; if((x == size) && (y == 0)) { for(int i = 0; i < size; i++) { for(int j = 0; j < size; j++) { cout << data[i][j] << " "; } cout << endl; } } } private: int size; int x, y; int **data; }; class tree_4 { public: int dVal; tree_4 *pFather; tree_4 *pChild; public: tree_4(tree_4 *father = NULL, int val = 0) : dVal(val), pFather(father), pChild(NULL) { } ~tree_4() { if(pChild) { delete [] pChild; pChild = NULL; } } void add_child(int val) { pChild = new tree_4[val]; for(int i = 0; i < val; i++) { pChild[i].dVal = 1 + i; pChild[i].pFather = this; } } void del_child() { if(pChild) { delete [] pChild; pChild = NULL; } } int depth() { if(pFather == NULL) return 0; return pFather->depth() + 1; } }; /* 遍历子树 */ void travel(tree_4 *pTree) { if(pTree == NULL) return; if(!current->push_back(pTree->dVal)) return; if(pTree->depth() == max_size * max_size) // 到达叶结点，找到一个符合要求的矩阵 { total_cnt ++; if(LOG_ON) current->show(); } else { pTree->add_child(max_size); for(int i = 0; i < max_size; i++) travel(&(pTree->pChild[i])); pTree->del_child(); } current->pop_back(); } void question_4(int val) { max_size = val; total_cnt = 0; current = new array_4(val); tree_4 source(NULL, 0); time_t start, end; start = time(NULL); source.add_child(max_size); for(int i = 0; i < max_size; i++) { travel(&(source.pChild[i])); } end = time(NULL); if(!LOG_ON) { cout << "/nCalcu time is: " << difftime(end, start) << "s;" << endl; } cout << "/nTotal latin maxtrix count is: " << total_cnt << endl; if(current) delete current; } void main() { question_4(5); }