LeetCode144. 二叉树的前序遍历

递归和非递归实现前序遍历

//144. 二叉树的前序遍历
//dfs，前序遍历模板
class Solution {
public:
  vector<int> ans;
  void dfs(TreeNode* root) {
    if (root == nullptr) return;
    ans.emplace_back(root->val);
    dfs(root->left);
    dfs(root->right);
  }
  vector<int> preorderTraversal(TreeNode* root) {
    vector<int>ans;
    dfs(root);
    return ans;
  }
};
// 进阶要求：尝试使用迭代解决
// 使用栈模拟实现dfs（使用队列模拟bfs）
class Solution {
public:
  vector<int> preorderTraversal(TreeNode* root) {
    vector<int>ans;
    if (root == nullptr) return ans;
    stack<TreeNode*>stk;
    auto node = root;
    while (stk.size() || node != nullptr) {
      while (node != nullptr) {//一直循环找node节点的左孩子
        ans.emplace_back(node->val);
        stk.push(node);
        node = node->left;
      }
      //找到右孩子，继续循环找左节点
      node = stk.top(); stk.pop();
      node = node->right;
    }
    return ans;
  }
};