二叉树的各种遍历(前中后非递归、层次、s形遍历)

之前去头条面试,凉在了二面的非递归二叉树遍历上,为避免再凉在同样的问题上,所以总结了一下。
二叉树有前序、中序、后序、层次、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;
}

关于二叉树先告一段落,等又有其他的再来补全。

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值