二叉树的遍历,递归和非递归

1,中序遍历

非递归版本,借助一个辅助

vector<int> ans;
vector<int> inorderTraversal(TreeNode* root) {
	TreeNode *node = root;
	stack<TreeNode *> mstack;
	while(node != NULL || !mstack.empty())
	{
	while(node != NULL)
	{
	    mstack.push(node); // 注意这里入栈不能写指针,写结点
	    node = node->left;
	}

		ans.push_back(mstack.top()->val); // 注意这里是 .
		node = mstack.top()->right;
		mstack.pop(); 
	}
	return ans;
}

递归版本,简洁

vector<int> ans;
vector<int> inorderTraversal(TreeNode* root) 
{
	if(root != NULL)
	{
	    inorderTraversal(root->left);
	    ans.push_back(root->val);
	    inorderTraversal(root->right);
	}
    
  return ans;
}

2,二叉树的后续遍历

非递归版本,后续遍历稍微麻烦点,通过设置一个 “前一个节点”变量,来记录下一次应该访问的结点

vector<int> ans;
vector<int> postorderTraversal(TreeNode* root) 
{
	if(root == NULL) 
		return ans;

	stack<TreeNode*> mstack;
	TreeNode* pre = NULL; //前一个节点
	while(!mstack.empty()|| root != NULL)
	{
		if(root != NULL)
		{
			mstack.push(root);
			root = root->left;
		}
		else
		{
			root=mstack.top();
			if(root->right == NULL || root->right == pre)
			{
				//如果右子树不存在,则接下来该访问根节点。
				mstack.pop();
				ans.push_back(root->val);
				pre = root;
				root = NULL;
			}
			else
			{
				//否则柚子树存在,就先访问右子树
				pre = NULL;
				root = root->right;
			}
		}
	}
	return ans;
}

递归版本,简洁

vector<int> ans;
vector<int> postorderTraversal(TreeNode* root) {
    if(root != NULL)
    {
        postorderTraversal(root->left);
        postorderTraversal(root->right);
        ans.push_back(root->val);
    }
    return ans;
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值