二叉樹非遞歸使用常數存儲空間遍歷

 

主要使用了一個方向值orien，記錄在二叉樹中搜索時的路徑方向，比如由父節點到左子樹時orien值爲1，這樣按照嚴格的行走規則，在一個while循環裏就能遍歷整棵樹。

    一個簡單的例子：

    一棵樹爲：            8

                  4             12

              2       6     10       14

             1 3     5 7   9  11   13  15

    那麼程序將採用一個相當於後序遍歷的順序對樹進行遍歷，對於一個節點，將先搜索左子樹，再搜索右子樹，然後輸出該節點的值。根據orien值的不同，對遍歷的下一步走向進行

    選擇，具體看printTree中的while循環。對於上面這棵樹，輸出順序爲：1 3 2 5 7 6 4 9 11 10 13 15 14 12 8。

    程序沒有測試一些特殊情況，可能存在bug，歡迎指正；若有啥問題歡迎交流！

#include <iostream> #include <cstdio> #include <cstring> using namespace std; int orien;//方向值， 1代表當前節點是某個節點的左子樹 // 2代表當前節點是某個節點的右子樹 // 3代表當前節點有一個左子樹，且左子樹剛剛被查看過 // 4代表當前節點有一個右子樹，且右子樹剛剛被查看過 class node{ public: int key; node* parent; node* left; node* right; node(int k=0) { key = k; parent = left = right = NULL; } }; class tree{ public: node* root; tree() { root = NULL; } void insert(node* n)//二叉查找樹的插入 { node* y=NULL; node* x = root; while (x != NULL) { y = x; if (n->key < x->key) x = x->left; else x = x->right; } n->parent = y; if (y == NULL) { root = n; } else if (n->key < y->key) { y->left = n; } else { y->right = n; } } }; void buildTree(tree* pt)//建立一棵樹 { int ar[]={8,4,12,2,6,10,14,1,3,5,7,9,11,13,15}, i; node* n; for (i=0; i<15; i++) { n = new node(ar[i]); pt->insert(n); } } node* findLeft(node* n)//查找最左節點 { while (n->left!=NULL || n->right!=NULL) { if (n->left==NULL) { n = n->right; orien = 2; } else { n = n->left; orien = 1; } } return n; } void printTree(const tree& t) { node* r = t.root; node* s; if (r == NULL) return; orien = 0; r = findLeft(r); while (r!=NULL || orien!=4) { if (orien == 1) { cout<<r->key<<" "; r = r->parent; orien = 3; } else if (orien == 2) { cout<<r->key<<" "; r = r->parent; orien = 4; } else if (orien == 3) { if (r->right == NULL) { s = r->parent; cout<<r->key<<" "; if (s == NULL){break;} if (r == s->left) orien = 3; else orien = 4; } else { s = r->right; orien = 2; s = findLeft(s); } r = s; } else if (orien == 4) { s = r->parent; cout<<r->key<<" "; if (s==NULL)break; if (s->left == r) orien = 3; else orien = 4; r = s; } } cout<<endl; } int main() { tree t; buildTree(&t); printTree(t); return 0; }