Leetcode[栈] 144. 二叉树的前序遍历

审题

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

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

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

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

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

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

提示：
树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100
 
进阶：递归算法很简单，你可以通过迭代算法完成吗？

看到这道题，一看，递归实现直接就可以AC了，然而为了掌握递归的底层逻辑，我们还是要搞一下迭代试一试

1. 在方案一中，递归。
2. 在方案二中，迭代（递归的本质就是模拟一个栈）。

代码实现

方案一：

常规操作，没啥好说的。

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

方案2：

这个迭代，我有一个地方还是每太理解，就是标注释的地方，这个条件让我楞想可能想不太出来…

1：这个的整体思路就是第一个循环，直接遍历到最左下，这个最左下可以认为是二叉树的最左下的左节点（一个空结点）。

2：然后将二叉树最左下的结点pop出来

3：遍历右结点，如果右结点为非空，继续转到1。如果右结点为空结点，直接将该最左下的父结点pop掉，再转回3，以此类推。

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

反思

这个栈的知识还是比较浅薄，尤其是结合了树之后。因此，如果想要掌握好这部分的内容，我认为首要的不是求甚解，而是不求甚解，在极其细节的部分可以先记下来，以后随着练习的增多自然会加深理解，不必搞自己心态。