欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Leetcode[树] 94. 二叉树的中序遍历

程序员文章站 2024-01-11 17:08:34
...

Leetcode[树] 94. 二叉树的中序遍历


Leetcode[树] 94. 二叉树的中序遍历

审题

给定一个二叉树的根节点 root ,返回它的中序遍历。

示例 1:
输入:root = [1,null,2,3]
输出:[1,3,2]

示例 2:
输入:root = []
输出:[]

示例 3:
输入:root = [1]
输出:[1]

示例 4:
输入:root = [1,2]
输出:[2,1]

示例 5:
输入:root = [1,null,2]
输出:[1,2]

中序遍历,对于递归算法来说很好想,也很好实现,但是本题要求的不是中序遍历输出元素,而是将元素保存在vector中,最后返回一个vector,因此,我们采用非递归遍历。

代码实现

方案一(上文提到的的方法):

这个和学二叉树的时候隔得时间太长了,到后来只能看回陈姥姥的PPT,小改了一下,抄了一段上来,不得不佩服人家这码力。

代码实现上,首先利用的是栈来模仿递归,因为要保存之前访问过的节点,整体的思路就是,遇到一个结点,就把它压栈,并去遍历它的左子树,当左子树遍历结束后,从栈顶弹出这个结点并访问它,然后按照其右指针去中序遍历该节点的右子树。(说人话)

好吧,其实就是采用了一个递归的思想, 首先遍历根节点的左子树(通过push(root)),怎么遍历根节点的左子树呢?通过遍历根节点的左子节点,怎么遍历根节点的左子节点呢?通过遍历根节点的左子节点的左子树->根->右子树…以此类推。

比如第一次循环,栈顶的元素应该是最左下的一个节点,因为没有左子树了,所以弹出他自己,然后遍历它的右子树…

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

方案2:
哦哦,看完官方题解后发现,这道题也可以用递归,直接把vector当参数传进去就行了…是我太菜呜呜呜、

直接一气呵成,over。

class Solution {
public:
    void inorderTraversal_para(TreeNode* root, vector<int>& v) {
        if ( root == nullptr ) return;
        inorderTraversal_para(root->left, v);
        v.push_back(root->val);
        inorderTraversal_para(root->right,v);
    }

    vector<int> inorderTraversal(TreeNode* root) {
       vector<int> v;
       inorderTraversal_para(root, v);
       return v;
    }
};

反思

树的遍历需要栈的相关知识和递归的用法,还挺综合的,不过对树的了解,我还是知之甚少,再看吧,再做两道题感受一下,今天先这样,收工。