(leetcode)二叉树的中序遍历

最新推荐文章于 2024-06-23 15:19:58 发布

原创最新推荐文章于 2024-06-23 15:19:58 发布 · 226 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #leetcode

leetcode数据结构专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用栈实现二叉树中序遍历的算法，详细展示了如何通过迭代方式而非递归来访问二叉树的所有节点，并将节点值存储到数组中。此方法避免了递归可能导致的栈溢出问题。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> a;
        if(root == NULL) return a;
        //判断树不为空
        if(root != NULL){
            //初始化栈
            stack<TreeNode*> s;
            //遍历辅助节点
            TreeNode *q;
            q = root;
            while(!s.empty()||q!=NULL){
                //左子树入栈
                while(q!=NULL){
                    s.push(q);
                    q = q->left;
                }
                //出栈节点有右孩子则入栈
                if(!s.empty()){
                    q = s.top();
                    a.push_back(q->val);
                    s.pop();
                    q = q->right;
                }
            }
        }
        return a;
    }
};