【LeetCode】-- 144. 二叉树的前序遍历

1. 题目

给你二叉树的根节点 root ,返回它节点值的 前序 遍历。

2. 示例

示例一:

【LeetCode】-- 144. 二叉树的前序遍历_第1张图片

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

 示例三:

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

示例4:

【LeetCode】-- 144. 二叉树的前序遍历_第2张图片

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

3. 分析

如果用二叉树的非递归做这道题:

(1)把一棵树拆成两个部分去访问:左路节点,左路节点上的右子树

【LeetCode】-- 144. 二叉树的前序遍历_第3张图片

 前序遍历就是根~左子树~根~左子树,左路节点访问完了,还剩左路节点上的右子树。

(2)访问左路节点时,同时入栈左路节点

【LeetCode】-- 144. 二叉树的前序遍历_第4张图片

(3)访问右子树时,先弹出左路节点,再访问它的右子树,不断拆解成子问题,用循环来做:

        1的左子树访问完了要访问1的右子树,1的右子树为空,还有3和5的右子树没访问,1作为3的左子树,3的左子树访问完了,要访问3的右子树,4和6访问完之后,要访问5的右树,7的左子树为空,访问完了就访问7的右子树8,8的左右子树都为空,结束。

【LeetCode】-- 144. 二叉树的前序遍历_第5张图片

 完整过程如下:

【LeetCode】-- 144. 二叉树的前序遍历_第6张图片 

 【LeetCode】-- 144. 二叉树的前序遍历_第7张图片

 【LeetCode】-- 144. 二叉树的前序遍历_第8张图片

 

 【LeetCode】-- 144. 二叉树的前序遍历_第9张图片

 【LeetCode】-- 144. 二叉树的前序遍历_第10张图片

 

【LeetCode】-- 144. 二叉树的前序遍历_第11张图片

【LeetCode】-- 144. 二叉树的前序遍历_第12张图片

 【LeetCode】-- 144. 二叉树的前序遍历_第13张图片

 

4. 代码实现

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector preorderTraversal(TreeNode* root) {
        stack st;
        vector ret;
        TreeNode* cur = root;

        while(cur || !st.empty())//cur指向哪个节点就表示前序访问这棵树
        {
            //1.访问左路节点,左路节点入栈,直到左路节点为空
            while(cur)
            {
                ret.push_back(cur->val);
                st.push(cur);
                cur = cur->left;
            }

            //2.依次取左路节点的右子树出来访问
            TreeNode* top = st.top();
            st.pop();

            //以子问题的形式访问这些右子树
            cur = top->right;
        }

        return ret;
    }
};

你可能感兴趣的:(LeetCode,C++,leetcode,算法)