二叉树的建立与遍历

二叉树的建立与遍历：

#include <iostream>
#include<string>
#include <vector>
#include<algorithm>
#include <stack>
#include <queue>
using namespace std;
//二叉树的建立与遍历
typedef struct TreeNode
{
char val;
struct TreeNode* left;
struct TreeNode* right;

}TreeNode;
void CreateTree(TreeNode* &root)
{
char data;
cin>>data;
if(data=='#')
root=NULL;
else
{
root=new TreeNode;
root->val=data;
CreateTree(root->left);
CreateTree(root->right);
}

}
void PreTree(TreeNode* root)
{
if(root)
{
cout<<root->val<<" ";
PreTree(root->left);
PreTree(root->right);
}

}
void InTree(TreeNode* root)
{
if(root)
{
InTree(root->left);
cout<<root->val<<" ";
InTree(root->right);
}

}
void PostTree(TreeNode* root)
{
if(root)
{
PostTree(root->left);
PostTree(root->right);
cout<<root->val<<" ";
}

}
/* 先序遍历(非递归)
思路：访问T->data后，将T入栈，遍历左子树；遍历完左子树返回时，栈顶元素应为T，出栈，再先序遍历T的右子树。
*/
void PreTree2(TreeNode* root)
{
stack<TreeNode*>s;
while(!s.empty()||root)
{
if(root)
{
s.push(root);
cout<<root->val<<" ";
root=root->left;
}
else
{
root=s.top();
s.pop();
root=root->right;
}

}

}
/* 中序遍历(非递归)
思路：T是要遍历树的根指针，中序遍历要求在遍历完左子树后，访问根，再遍历右子树。
先将T入栈，遍历左子树；遍历完左子树返回时，栈顶元素应为T，出栈，访问T->data，再中序遍历T的右子树。
*/
void InTree2(TreeNode* root)
{
stack<TreeNode*>s;
while(!s.empty()||root)
{
if(root)
{
s.push(root);
root=root->left;

}
else
{
root=s.top();
cout<<root->val<<" ";
s.pop();
root=root->right;
}

}

}
void PostTree2(TreeNode* root)
{
stack<TreeNode*>s;
TreeNode* preroot=NULL;
while(!s.empty()||root)
{
while(root)//一直左走到底
{
s.push(root);
root=root->left;
}
root=s.top();
if(root->right==NULL||root==preroot) // 当前节点的右孩子如果为空或者已经被访问，则访问当前节点
{
cout<<root->val<<" ";
s.pop();
root=NULL;
}
else
{
preroot=root;
root=root->right;
}

}
}
//层次遍历
void LevelTree(TreeNode* root)
{
queue<TreeNode*> q;
if(root)
q.push(root);
while(!q.empty())
{
root=q.front();
cout<<root->val<<" ";
q.pop();
if(root->left) q.push(root->left);
if(root->right)q.push(root->right);
}
}
int main()
{

TreeNode* root;
CreateTree(root);
cout<<"先序遍历递归："<<endl;
PreTree(root);
cout<<endl;
cout<<"中序遍历递归："<<endl;
InTree(root);
cout<<endl;
cout<<"后序遍历递归："<<endl;
PostTree(root);
cout<<endl;

cout<<"先序遍历非递归："<<endl;
PreTree2(root);
cout<<endl;
cout<<"中序遍历非递归："<<endl;
InTree2(root);
cout<<endl;
cout<<"后序遍历非递归："<<endl;
PostTree2(root);
cout<<endl;
cout<<"层次遍历非递归："<<endl;
LevelTree(root);
cout<<endl;

return 0;
}