二叉树前序遍历（迭代）

S1eepzZ

于 2024-07-07 11:05:25 发布

阅读量255

点赞数 4

文章标签：算法数据结构

本文链接：https://blog.csdn.net/S1eepzZ/article/details/140243171

版权

二叉树前序遍历：根节点，左子树，右子树的顺序来遍历二叉树迭代的方式就套模板（中序遍历）就行，这样比较容易记忆。

/**
 * 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> vec;
        stack<TreeNode*> s;
        while(root != nullptr || !s.empty()){
            while(root != nullptr){
                vec.push_back(root->val);
                s.push(root);
                root = root ->left;
            }
            TreeNode* t = s.top();
            root = t->right;
            s.pop();
        }
        return vec;
    }
};

前面的两个while循环就是模板，可以直接套，比较容易记忆。

因为前序遍历是按照根节点，左子树，右子树的顺序所以在要先进行vec.push_back(root->val)后压栈。

Stack的作用就是让遍历可以往回走（个人理解）