力扣 144. 二叉树的前序遍历

前序遍历的节点顺序为，中->左->右。本题使用了迭代法和递归法两种方法。递归和迭代都是循环的一种。 简单地说，递归是重复调用函数自身实现循环，迭代是函数内某段代码实现循环。而迭代与普通循环的区别是，迭代的循环代码中参与运算的变量同时是保存结果的变量，当前保存的结果作为下一次循环计算的初始值。

迭代法是使用的堆栈来模仿递归遍历算法中通常使用的函数调用堆栈。因为栈是后进先出的数据结构，因此会先压入右子节点，再压入左子节点。

递归法的关键点有三点，分别是确定递归函数的参数和返回值，确定终止条件，确定单层递归的逻辑。

迭代法时间复杂度：O(n)，空间复杂度：O(n)

迭代法思路：

1. 使用堆栈来模仿递归遍历算法中通常使用的函数调用堆栈。
2. 首先检查给定的根节点是否为一棵空树。如果为空，则函数直接返回空结果。
3. 如果树不为空，则将根节点压入堆栈以启动遍历过程。前序遍历中，根节点是第一个遍历的节点。
4. 创建一个循环根据前序遍历的反顺序进行压栈出栈，直到栈没有节点压入为空。
   1. 从堆栈中弹出顶部节点。该节点代表当前正在处理的节点，即前序遍历中访问的当前节点。将当前节点的值添加到result向量中。
   2. 根据当前节点，检查是否有右子节点。如果是，则将右子节点压入堆栈。检查当前节点是否有左子节点。如果是，则将左孩子压入堆栈。之所以先压入右子节点，因为栈是后进先出（LIFO）数据结构，因此左子节点将更早的弹出，符合前序遍历的顺序。
5. 返回结果向量。

迭代法代码：

/**
 * 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<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;    // 使用堆栈来模仿递归遍历算法中通常使用的函数调用堆栈
        vector<int> result;
        if (root == NULL)   // 遇到空树，直接返回
            return result;
        st.push(root);
        while (!st.empty())
        {
            TreeNode* node = st.top();  // 第一次循环弹出根节点，之后每次循环就取栈最顶的节点，
            st.pop();
            result.push_back(node->val);
            if (node->right)    // 先压右节点，栈是先进后出
                st.push(node->right);
            if (node->left)     // 再压左节点
                st.push(node->left);
        }
        return result;
    }
};

递归法时间复杂度：O(n)，空间复杂度：O(n)

递归法思路：

1. 在本题的前序遍历中，创建一个vector<int>类型的参数，在该vector中存放遍历节点的数值，最后返回。
2. 确定递归函数的参数和返回值：在递归函数中，因为要存放前序遍历节点的数值，所以参数里需要传入vector的指针或者引用来放节点的数值，不需要返回值，所以递归函数返回类型为void。
3. 确定终止条件：在递归函数中，如果当前遍历的这个节点是个空节点，则直接return。
4. 确定单层递归的逻辑：前序遍历是中左右的循序，所以在递归函数的逻辑中，先取中节点的数值，再递归取左节点的数值，再递归取右节点的数值。
5. 返回存放结果vector<int>。

递归法代码：

/**
 * 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:
    void traversal(TreeNode *node, vector<int> &ret)
    {
        if (node == NULL)
            return;
        ret.push_back(node->val);
        traversal(node->left, ret);
        traversal(node->right, ret);
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> ret;
        traversal(root, ret);
        return ret;
    }
};