LeetCode 144. Binary Tree Preorder Traversal

最新推荐文章于 2024-05-30 07:40:02 发布

原创最新推荐文章于 2024-05-30 07:40:02 发布 · 171 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 题解专栏收录该内容

288 篇文章

订阅专栏

本文介绍了一种使用栈实现二叉树节点值的前序遍历的方法，并提供了一个迭代器类来辅助遍历过程。

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

// Given a binary tree, return the preOrder traversal of its nodes' value
// The main idea here is to use stack, nothing special

vector<int> preorderTraversal(TreeNode* root) {
    vector<int> result;
    if(!root) return result;
    stack<TreeNode*> nodes;
    nodes.push(root);
    while(!nodes.empty()) {
        TreeNode* tmp = nodes.top();
        nodes.pop();                     // remember to pop it out.
        result.push_back(tmp->val);
        if(tmp->right != NULL)
            nodes.push(tmp->right);
        if(tmp->left != NULL)
            nodes.push(tmp->left);
    }
}

Preorder iterator

class PreorderIterator {
private:
  stack<TreeNode*> nodes;

public:
  void preorderIterator(TreeNode* root) {
    nodes.push_back(root);
  }
  void hasNext() {
    return !nodes.empty();
  }

  void getNext() {
    if(hasNext()) {
      TreeNode* top = nodes.top();
      if(top->right) nodes.push_back(top->right);
      if(top->left) nodes.push_back(top->left);
      nodes.pop();
    }
  }
}