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$
  

思路

1. 思路一 ：使用递归遍历

这是一个关于二叉树的经典问题，我们可以使用递归的方法对其进行深度优先的遍历方式。本题是前序遍历，因此需要按照根，左，右的顺序。

如上图所示，进行前序遍历的顺序也就是 $54136$ 。
在使用递归中需要确认三个要素。

• 递归函数的参数，返回值
• 递归函数的终止条件
• 递归函数每一层需要的操作

在这个递归中需要访问每个节点一遍，因此当前的节点是参数之一，同时需要在每次递归开始时访问当前字子树的根节点，因此需要传入一个数组保存当前被访问结点的值。
我们需要在节点为空时终止访问，因此需要将终止条件设为节点为空防止出现栈溢出的现象。
最后需要判定每一层需要做的操作，先访问当前子树的根节点，接着进入左子树，最后是右子树。
代码：

class Solution {
    private:
    vector<int> list;

    public:
    vector<int> preorderTraversal(TreeNode* root) {
        
        perorderRecursion(root,list);
        return list;
    }

    void perorderRecursion(TreeNode* cur,vector<int>& list){
        
        //确认递归终止条件
        if(cur == nullptr){
            return;
        }
        
        list.push_back(cur -> val);
        perorderRecursion(cur -> left,list);
        perorderRecursion(cur -> right,list);
    }

};

2. **思路二 ：使用迭代遍历 **

对于递归遍历我们的思路是很清楚的，那么有没有可以不使用递归的方法呢？
我们首先需要想到递归的实现原理，递归的过程中每一次调用将函数的局部变量，参数值和返回地址等压入栈中，在调用结束时从栈顶弹出上次递归的各项参数和值。因此我们可以使用栈这个数据结构来替代递归对于遍历二叉树的处理。

在进行迭代遍历中，我们首先访问根节点，同时对根节点进行入栈操作。

按照前序遍历的顺序，对根节点进行操作，将其弹出栈放入数组中，同时将其子节点按照右子节点，左子节点的顺序进行入栈操作。之所以按照先右后左的顺序是因为栈是先入后出的，因此在后面出栈的时候就成了根左右的顺序了。

按照这个方式直到栈中的元素全部为空。

代码：

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> stack;
        vector<int> vector;

        if(root != nullptr){
            stack.push(root);
        }

        while(stack.empty() == false){
            TreeNode* cur = stack.top();
            stack.pop();

            vector.push_back(cur -> val);
            
            if(cur -> right != nullptr){
                stack.push(cur -> right);
            }

            if(cur -> left !=nullptr){
                stack.push(cur -> left);
            }

        }
        return vector;
    }
};