leetcode 二叉树的遍历栈的实现 Preorder Traversal Inorder Traversal Postorder Traversal

题目意思是二叉树的三种遍历方式,不用递归用栈实现,之前数据结构中做过。
先序遍历:大概就是每次都pop一个,然后加右孩子,左孩子。

Given a binary tree, return the preorder traversal of its nodes' values.

For example:
Given binary tree {1,#,2,3},

   1
    \
     2
    /
   3

return [1,2,3].

Note: Recursive solution is trivial, could you do it iteratively?

 class Solution {
 public:
	 vector<int> preorderTraversal(TreeNode* root) {
		 vector<int> res;
		
		 if(!root) return res;
		 stack<TreeNode*> S;
		 S.push(root);
		 TreeNode* node;
		 while (!S.empty())
		 {
			 node = S.top();
			 S.pop();
			 res.push_back(node->val);
			 if (node->right) S.push(node->right);
			 if (node->left) S.push(node->left);
		 }
		 return res;
	 }
 };

中序遍历:
 class Solution {
 public:
	 vector<int> inorderTraversal(TreeNode* root) {
		 vector<int>res;
		 if (!root) return res;
		 stack<TreeNode *>S;
		 TreeNode* node=root;
		 while (node||!S.empty())
		 {
			 while (node)//有左就一直向左
			 {
				 S.push(node);
				 node = node->left;
			 }
			 node = S.top();
			 S.pop();
			 res.push_back(node->val);
			 node = node->right;
		 }
		 return res;
	 }
 };

后序遍历:大概就是有左右子树就push,没有左子树和右子树或者已经处理过了就输出,这里已经处理过了可以用一个child节点记录上一个输出的node,若这个child是栈顶的左孩子或者右孩子就表示已经处理过了。
class Solution {
 public:
	 vector<int> postorderTraversal(TreeNode* root) {
		 vector<int>res;
		 if (!root) return res;
		 stack<TreeNode*>S;
		 S.push(root);
		 TreeNode* node = root;
		 TreeNode* child = node;
		 while (!S.empty())
		 {
			 node = S.top();
			 if ((!node->left&&!node->right) || child == node->left || child == node->right)//左右都空了或者孩子已经处理完成了
			 {
				 res.push_back(node->val);
				 S.pop();
				 child = node;
			 }
			 else{ //先把右孩子push,再把左孩子push
				 if (node->right) S.push(node->right);
				 if (node->left)  S.push(node->left);
			 }
		 }
		 return res;
	 }
 };



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值