【Leetcode】144. Binary Tree Preorder Traversal-CSDN博客

本文链接：https://blog.csdn.net/qq_46105170/article/details/104026868

本文介绍四种二叉树先序遍历的方法，包括深度优先搜索、分治法、栈模拟递归及改进版栈法。每种方法均有详细代码示例与时间空间复杂度分析，以及算法正确性的数学归纳法证明。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：

https://leetcode.com/problems/binary-tree-preorder-traversal/

先序遍历二叉树，并存入一个list里。

法1：DFS，代码如下：

/**
 * 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) {
    vector<int> res;
    dfs(root, res);
    return res;
  }

  void dfs(TreeNode* root, vector<int>& res) {
    if (!root) return;
    res.push_back(root->val);
    dfs(root->left, res);
    dfs(root->right, res);
  }
};

时间复杂度 $O (n)$ ，空间 $O (h)$ ， $h$ 为二叉树深度，也就是递归栈深度。

法2：用栈模拟递归。从树根开始，每次遍历左链，同时将非空右孩子入栈。左链遍历完毕之后从栈里弹一个元素，接着重复上述操作。代码如下：

class Solution {
 public:
  vector<int> preorderTraversal(TreeNode* root) {
    vector<int> res;
    stack<TreeNode*> stk;
    auto cur = root;
    while (cur || stk.size()) {
      while (cur) {
        res.push_back(cur->val);
        if (cur->right) stk.push(cur->right);
        cur = cur->left;
      }
      if (stk.size()) {
        cur = stk.top();
        stk.pop();
      }
    }
    return res;
  }
};

时空复杂度和之前一样。