144. 二叉树的前序遍历-C++

题目来源:力扣

示例 1:

144. 二叉树的前序遍历-C++_第1张图片

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

示例 2:

输入:root = []
输出:[]

代码实现:

class Solution {
public:
    vector preorderTraversal(TreeNode* root) {
        stack st;
        TreeNode* cur = root;
        vector v; 
        while(cur || !st.empty())
        {
            //访问一棵树的开始
            //1.访问左路节点,左路节点入栈
            while(cur)
            {
                v.push_back(cur->val);
                st.push(cur);
                cur=cur->left;
            }
            //依次访问左路节点的右子树
            TreeNode* top = st.top();
            st.pop();
            //字问题的方式访问右子树
            cur = top->right;
        }
        return v;
    }
};

 思路:

144. 二叉树的前序遍历-C++_第2张图片

我们用这棵树来举例

144. 二叉树的前序遍历-C++_第3张图片 

我们在访问一棵树,按照前序遍历,我们最先访问的是左路节点,也就是8,3,1,所以我们就可以把树分为两个部分,一个是左路节点,另一个就是左路节点的右子树,我们把左路节点入栈

然后我们取栈顶元素,也就是1,我们访问完了1,然后pop一下,接着访问1的右子树,1的右子树为空,不用管

144. 二叉树的前序遍历-C++_第4张图片 接着栈顶元素就是3了,我们访问3,然后pop一下,再访问3的右子树,也就是6,按照刚才的规则,把6和4入栈,接着就是4被访问,pop掉,6被访问,pop掉,访问6的右子树,也就是7,继续,7入栈,访问7,pop掉,然后7的左右为空,不用管

此时栈中就只剩8了,然后我们访问8,pop掉,再访问8的右子树,这个结果就是前序遍历了

 我们要注意的是,cur指向一个节点代表访问一棵树的开始,栈里面的节点代表要依次访问它的右子树,大家可以画图理解一下

你可能感兴趣的:(c++,算法,开发语言,leetcode,数据结构)