转载请注明t1234xy4 原创 http://blog.csdn.net/t1234xy4/article/details/52388179
二叉树遍历简介
前面已经写过一篇关于二叉树的遍历,但看到一个更好的遍历方法:Morris遍历。
二叉树的递归遍历:主要是通过函数调用栈从子树返回到父亲节点,非递归的方法是通过额外空间栈来保存子节点返回父亲节点。
如果树的深度够高,递归可能造成函数栈溢出。而非递归的需要使用栈,有O(lgn)的空间消耗。
Morris遍历是通过时间换空间的方法,时间复杂度为O(n*2),后序时间复杂度约为O(n*4),空间复杂度为O(1)。
Morris遍历控制条件
我们设置Cur为当前节点,pre为查找指针。
初始化:
Cur = root;
控制条件:
[1] Cur!=NULL; //说明没有遍历完所有的节点;
<pre name="code" class="cpp">[2] 设置pre = Cur->left; //即指pre的左子树为空;
[3] pre->right == NULL ;//为当前节点前驱节点
[4] pre->right == Cur ; //前驱节点
Morris前序遍历
1、控制条件[1],保证所有节点都被遍历; 我们依次查找当前Cur节点的前驱,将前驱节点的right指针指向Cur,以便于下次从子树返回父亲节点用。
2、控制条件[2],每当当前节点Cur没有左子树时,下一步应该输出当前节点,并且该进入当前节点的右子树,如果右子树存在,就进入了右子树,否则进入了父亲节点。
3、控制条件[3],以便于找到当前节点的前驱,为了能从子树返回到父亲节点,此处应该将它修改为指向当前节点。
4、控制条件[4],说明当前节点的左子树都已经遍历过了,这时候应该返回到父节点,这时候需要将前驱节点的right 设置为 null pre->right = null。
void BinaryTreeMorrisDataFirst(BinaryNode* root){
if(root==NULL) return;
BinaryNode* Cur = root;
BinaryNode* pre = NULL;
while(Cur!=NULL){
pre = Cur->left;
while (pre!=NULL && pre->right!=NULL && pre->right!=Cur)
pre = pre->right;
if(pre==NULL){
printf("%4d",Cur->value);
Cur = Cur->right;
continue;
}
if(pre->right == NULL){
printf("%4d",Cur->value);
pre->right = Cur;
Cur = Cur ->left;
}else{
pre->right = NULL;
Cur = Cur->right;
}
}
}
Morris中序
与前驱节点的控制方法一样,只是输出位置不同,中序应该
pre = null 时,即当前节点没有左子树,输出当前节点;Cur = Cur->right (返回父亲节点 或者 进入右子树,如果存在右子树)
pre->right == Cur时,即左子树遍历完,此时该遍历当前节点,并Cur = Cur->right.
pre->right==null 说明第一次进入,此时应该为返回做好准备,设置pre->right =Cur,并让Cur =Cur->left。
void BinaryTreeMorrisMiddle(BinaryNode* root){
if(root==NULL) return;
BinaryNode* Cur = root;
BinaryNode* pre = NULL;
while(Cur!=NULL){
pre = Cur->left;
while(pre!=NULL && pre->right!=NULL&& pre->right!=Cur)
pre = pre->right;
if(pre==NULL){
printf("%4d",Cur->value);
Cur = Cur->right;
continue;
}
if(pre->right==NULL){
pre->right = Cur;
Cur = Cur->left;
continue;
}
if (pre->right == Cur)
{
printf("%4d",Cur->value);
Cur=Cur->right;
pre->right =NULL;
}
}
}
Morris后序遍历
把遍历看成是反斜杠 \ 逆序输出。
同样前面的控制条件。
1、每次逆序输出当前的Cur->left 到 Cur前驱节点这一条链上的所有节点。
2、为了逆序不开空间,需要逆转指针,输出后再逆转。
void BinaryTreeMorrisDataLast(BinaryNode* root){
if(root==NULL) return;
BinaryNode* Cur = root;
BinaryNode* pre = NULL;
while(Cur!=NULL){
pre = Cur->left;
while(pre!=NULL && pre->right!=NULL && pre->right != Cur)
pre = pre->right;
if(pre == NULL){
Cur = Cur->right;
continue;
}
if(pre->right == NULL){
pre->right = Cur;
Cur = Cur->left;
}
if(pre->right == Cur){
pre->right = NULL;
reverseAndprint(Cur->left);
Cur = Cur->right;
}
}
reverseAndprint(root);
}
Morris遍历代码分享
#include <iostream>
#include <algorithm>
using namespace std;
struct BinaryNode{
BinaryNode* left;
BinaryNode* right;
int value;
public:
BinaryNode(int x):value(x),left(NULL),right(NULL){}
};
/************************************************************************/
/* to be test: null / 1 2 3 4 */
/* 6,5,2,1,4,3,5,7,7,9,8,10 */
/************************************************************************/
void BinaryTreeInsert(BinaryNode** root,int key){
if(root == NULL) return;
if(root!=NULL && (*root)==NULL){
(*root) = new BinaryNode(key);
return;
}
BinaryNode *pNode = (*root);
BinaryNode* pre = pNode;
while(pNode!=NULL){
pre = pNode;
pNode = (key < pNode->value)?pNode->left:pNode->right;
}
BinaryNode* node = new BinaryNode(key);
(key<pre->value)?(pre->left = node):(pre->right = node);
}
/************************************************************************/
/* to be test:null // 1 2 3 4 //6,5,2,1,4,3,5,7,7,9,8,10 */
/************************************************************************/
void BinaryTreeMorrisMiddle(BinaryNode* root){
if(root==NULL) return;
BinaryNode* Cur = root;
BinaryNode* pre = NULL;
while(Cur!=NULL){
pre = Cur->left;
while(pre!=NULL && pre->right!=NULL&& pre->right!=Cur)
pre = pre->right;
if(pre==NULL){
printf("%4d",Cur->value);
Cur = Cur->right;
continue;
}
if(pre->right==NULL){
pre->right = Cur;
Cur = Cur->left;
continue;
}
if (pre->right == Cur)
{
printf("%4d",Cur->value);
Cur=Cur->right;
pre->right =NULL;
}
}
}
void LDR(BinaryNode* root){
if(root != NULL){
LDR(root->left);
printf("%4d",root->value);
LDR(root->right);
}
}
void BinaryTreeMorrisDataFirst(BinaryNode* root){
if(root==NULL) return;
BinaryNode* Cur = root;
BinaryNode* pre = NULL;
while(Cur!=NULL){
pre = Cur->left;
while (pre!=NULL && pre->right!=NULL && pre->right!=Cur)
pre = pre->right;
if(pre==NULL){
printf("%4d",Cur->value);
Cur = Cur->right;
continue;
}
if(pre->right == NULL){
printf("%4d",Cur->value);
pre->right = Cur;
Cur = Cur ->left;
}else{
pre->right = NULL;
Cur = Cur->right;
}
}
}
void DLR(BinaryNode* root){
if(root!=NULL){
printf("%4d",root->value);
DLR(root->left);
DLR(root->right);
}
}
void reverseAndprint(BinaryNode* head){
if(head==NULL) return;
BinaryNode* next = head;
BinaryNode* rev = NULL;
while(next!=NULL){
BinaryNode* p = next->right;
next->right = rev;
rev = next;
next = p;
}
next = rev;
rev = NULL;
while(next!=NULL){
BinaryNode* p = next->right;
printf("%4d",next->value);
next->right = rev;
next = p;
}
}
void BinaryTreeMorrisDataLast(BinaryNode* root){
if(root==NULL) return;
BinaryNode* Cur = root;
BinaryNode* pre = NULL;
while(Cur!=NULL){
pre = Cur->left;
while(pre!=NULL && pre->right!=NULL && pre->right != Cur)
pre = pre->right;
if(pre == NULL){
Cur = Cur->right;
continue;
}
if(pre->right == NULL){
pre->right = Cur;
Cur = Cur->left;
}
if(pre->right == Cur){
pre->right = NULL;
reverseAndprint(Cur->left);
Cur = Cur->right;
}
}
reverseAndprint(root);
}
void LRD(BinaryNode* root){
if(root!=NULL){
LRD(root->left);
LRD(root->right);
printf("%4d",root->value);
}
}
int main(){
int m[10] = {6,2,1,4,3,5,7,9,8,10};
BinaryNode* root = NULL;
for(int i=0;i<10;i++)
BinaryTreeInsert(&root,m[i]);
LDR(root);
cout <<endl<<"Morris Middle:"<<endl;
BinaryTreeMorrisMiddle(root);
cout <<endl;
DLR(root);
cout << endl<<"Morris Data First:"<<endl;
BinaryTreeMorrisDataFirst(root);
cout<<endl ;
LRD(root);
cout<<endl<<"Morris Data Last:"<<endl;
BinaryTreeMorrisDataLast(root);
}
测试结果
原插入的树为: 6 2 1 4 3 5 7 9 8 10