力扣刷题记录36.1-----144. 二叉树的前序遍历(迭代实现)

@白圭

已于 2022-09-02 11:14:40 修改

阅读量147

点赞数

分类专栏： # 力扣习题记录册__C++__代码随想录文章标签： leetcode 算法 c++

于 2022-09-01 13:00:00 首次发布

本文链接：https://blog.csdn.net/taiyuezyh/article/details/126641098

版权

力扣习题记录册__C++__代码随想录专栏收录该内容

174 篇文章 0 订阅

订阅专栏

一、题目

在这里插入图片描述

二、代码

/**
 * 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 tree_for_each_recursive(TreeNode* cur,vector<int>& result)   //二叉树的递归遍历  其实是用的是栈
    {
        if(cur==nullptr) return;
        //中序和后序遍历只需要微调顺序即可
        result.push_back(cur->val);
        tree_for_each_recursive(cur->left,result);
        tree_for_each_recursive(cur->right,result);

    };

    vector<int> preorderTraversal(TreeNode* root) 
    {
      vector<int> return_vector;

      ///递归
      //tree_for_each_iteration(root,return_vector);
 
      //迭代
      stack<TreeNode*> my_stack_for_iteration;

      if(root==nullptr) return return_vector;
      else my_stack_for_iteration.push(root);  //先将根节点压入堆栈
      //思路:弹出处理节点 压入新节点 直到栈中无节点为止
     while(!my_stack_for_iteration.empty())
     {
        TreeNode* node=my_stack_for_iteration.top();  //有必要记录一下这个节点 不然弹出之后就找不到左右节点了
        return_vector.push_back(node->val);  //取出栈顶的元素   核心:下一次pop出来的是要处理的节点
        my_stack_for_iteration.pop();
        if(node->right) my_stack_for_iteration.push(node->right);
        if(node->left) my_stack_for_iteration.push(node->left);
     }
      return return_vector;
    }
};