leetcode学习日记7-二叉树的中序遍历

最新推荐文章于 2024-04-20 00:41:24 发布

weixin_44288450

最新推荐文章于 2024-04-20 00:41:24 发布

阅读量88

点赞数

本文链接：https://blog.csdn.net/weixin_44288450/article/details/100588170

版权

leetcode学习日记7-二叉树的中序遍历

题目如下

给定一个二叉树，返回它的中序遍历。

首先，什么是中序遍历呢？
遍历(Transversal)是对二叉树的每一个节点进行一次访问。
对于一个二叉树，有三种遍历方式。
在这里插入图片描述
左根右，即245
根左右，即425
左右根，即254
其中，第一种遍历方式就是中序遍历。

如何实现二叉树的中序遍历呢？
利用栈的思想。

比如现在有一个节点p，根据中序遍历的规则，首先需要遍历其左子树上的节点，然后遍历他自己，最后遍历他右子树上的节点。
这就好比先将他自己入栈，然后将他所有左子树上的节点入栈，然后将左子树上的节点出栈，他自己出栈，他右子树上的节点入栈，这些节点出栈。这样就完成了一个从父节点出发的中序遍历。

例子：
在这里插入图片描述

首先将根节点入栈
ans=[]
在这里插入图片描述
然后，遍历其左子树，入栈
ans=[]

在这里插入图片描述

左子树遍历结束，左子树出栈
ans=[2]
在这里插入图片描述

根节点出栈
ans=[2,4]
在这里插入图片描述

右子树入栈
ans=[2,4]
在这里插入图片描述

右子树的左子树入栈

在这里插入图片描述

右子树的左子树遍历完毕，左子树出栈
ans=[2,4,3]
在这里插入图片描述

右子树的父节点出栈
ans=[2,4,3,5]
在这里插入图片描述

右子树的右子树入栈
在这里插入图片描述

右子树的右子树出栈
ans=[2,4,3,5,6]
在这里插入图片描述

此时，栈为空且该节点没有可以继续遍历的下一个子树，故遍历完毕。
输出ans即为中序遍历。

代码如下

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        stack<TreeNode *> Stack;
        vector<int> res;
        TreeNode * p=root;
           while(!Stack.empty()||p!=NULL)
           {
             while(p!=NULL)
            {
              Stack.push(p);
              p=p->left;
             }
            
              p=Stack.top();
              Stack.pop();
              res.push_back(p->val);
              p=p->right;
           }
        return res;
        
    }