之前去头条面试,凉在了二面的非递归二叉树遍历上,为避免再凉在同样的问题上,所以总结了一下。
二叉树有前序、中序、后序、层次、s形等等,前三种有用递归实现(下面就不实现了,很简单),非递归实现。其实层次遍历就是广度优先遍历(BFS),深度优先遍历就是前中后序遍历。下面分别来实现一下。
二叉树定义
typedf struct node{
int data;
struct node* lchild;
struct node* rchild;
}BTNode
中序遍历(左 中 右)
对于上面的二叉树,打印顺序是:4 2 5 1 3
思路
要先打印左子树,再打印根节点,最后打印右子树,则要先到达左子树最左边的叶子结点,打印了该叶子结点后,再退回去打印对应的父节点,和对应的右孩子节点等等。如何实现呢,用栈。
void inOrder(BTNode* root){
if(root==NULL) return;
BTNode* p=root;
stack<BTNode*> s;
while(p!=NULL || !s.empty()){
while(p!=NULL){
s.push(p);
p=p->lchild;
}
if(!s.empty()){//p为空,说明已经没有左孩子了
p=s.top();
s.pop();
cout<<p->data;
p=p->rchild;
}
}
return;
}
前序遍历(中 左 右)
对于上面的二叉树,打印顺序是:1 2 4 5 3
思路
先打印根节点,将根节点入栈,再入栈左孩子,当没有左孩子了,就出栈打印栈顶元素,并开始指向其右孩子。
其实和中序一样,只是打印顺序不一样。
void preOrder(BTNode* root){
if(root==NULL) return;
BTNode* p=root;
stack<BTNode*> s;
while(p!=NULL || !s.empty()){
while(p!=NULL){
cout<<p->data;
s.push(p);
p=p->lchild;
}
if(!s.empty()){
p=s.top();
s.pop();
p=p->rchild;
}
}
return;
}
后序遍历(左 右 中)
对于上面的二叉树,打印顺序是:4 5 2 3 1
思路
同样可以用栈,传入根节点后,将左孩子入栈,一直到最左边的孩子入栈,开始打印。当打印的时候要判断(因为不知道是不是父节点),只有当前节点没有右孩子或者右孩子已经被打印过了,才打印;若有右孩子或者右孩子还没打印过,则将当前节点再次入栈(因为这个节点就是父节点,要在右孩子打印了之后才能被打印,所以要再次入栈,以便之后访问),则指向右孩子,并将右孩子的左孩子入栈。
void postOrder(BTNode* root){
if(root==NULL) return;
BTNode* p=root;
stack<BTNode*> s;
BTNode* cur,lastVisited;//cur表示当前节点,lastVisited记录上一次访问过的节点
cur=root;
lastVisited=NULL;
while(cur){
s.push(cur);
cur=cur->lchild;
}
while(!s.empty()){
cur=s.top();
s.pop();
if(cur->rchild==NULL || cur->rchild==lastVisited){
cout<<cur->data;
lastVisited=cur;
}
else{
s.push(cur);
cur=cur->rchild;
while(cur){
s.push(cur);
cur=cur->lchild;
}
}
}
return;
}
层次遍历
对于上面的二叉树,打印顺序是:1 2 3 4 5
思路
先打印根节点,再打印根节点对应的左右孩子,再打印前面左右孩子对应的左右孩子。可以用队列实现,每当要打印一个节点的时候,将它的左右孩子入队列,一次打印队列中的节点。
void levelOrder(BTNode* root){
if(root==NULL) return;
queue<BTNode*> q;
BTNode* p=root;
q.push(p);
while(!q.empty()){
p=q.front();//获取对首元素
cout<<p->data;
q.pop();
if(p->lchild) q.push(p->lchild);
if(p->rchild) q.push(p->rchild);
}
return;
}
蛇形遍历(leetcode 103)
对于上面的二叉树,打印顺序是:1 3 2 4 5
思路
用两个栈!一个栈存储从左往右的顺序节点,另一个栈存储从右往左的。
下面是leetcode上提交的程序,和前面的有些不一致,比如节点的定义,返回类型。
vector<vector<int>> zigzigLevelOrder(TreeNode* root){
vector<vector<int>> res;
if(!root) return res;
stack<TreeNode*> s1,s2;//s1记录从左往右(包含根节点),s2纪录从右往左
s1.push(root);
while(!s1.empty() || !s2.empty()){
vector<int> tmp;
if(!s1.empty()){
while(!s1.empty()){
TreeNode* cur=s1.top();
s1.pop();
if(cur->left) s2.push(cur->left);
if(cur->right) s2.push(cur->right);
tmp.push_back(cur->val);
}
}
else{
while(!s2.empty()){
TreeNode* cur=s2.top();
s2.pop();
if(cur->right) s1.push(cur->right);
if(cur->left) s1.push(cur->left);
tmp.push_back(cur->val);
}
}
res.push_back(tmp);
}
return res;
}
关于二叉树先告一段落,等又有其他的再来补全。