107. Binary Tree Level Order Traversal II

最新推荐文章于 2021-08-10 13:02:56 发布

原创最新推荐文章于 2021-08-10 13:02:56 发布 · 178 阅读

0 ·

CC 4.0 BY-SA版权

leetcode-二叉树专栏收录该内容

9 篇文章

订阅专栏

本文介绍了两种实现二叉树层次遍历的方法：一种使用栈（stack），另一种使用队列（queue）。通过这两种数据结构，我们可以从底部向上进行层次遍历，并获取每一层节点的值。

可以使用stack

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        stack<vector<TreeNode*>> st;
        vector<vector<int>> vt;
        
        if(root==NULL)
            return vt;
        
        vector<TreeNode*> v;
        v.push_back(root);
        st.push(v);
        
        while(1)
        {
            v=st.top();
            vector<TreeNode*> vs;
            for(int i=0;i<v.size();i++)
            {
                if(v[i]->left)
                    vs.push_back(v[i]->left);
                if(v[i]->right)
                    vs.push_back(v[i]->right);
            }
            if(vs.size()==0)
                break;
            st.push(vs);
        }
    
        while(!st.empty())
        {
            v=st.top();
            st.pop();
            vector<int> tmp;
            for(int j=0;j<v.size();j++)
            {
                tmp.push_back(v[j]->val);
            }
            vt.push_back(tmp);
        }
     
        return vt;
    }
};

2.使用queue

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>> res;
        queue<TreeNode*> vec;

        if(root==NULL)
            return res;
        
        vec.push(root);
        
        while(!vec.empty())
        {
            queue<TreeNode*> tmp_vec;
            vector<int> tmp_res;
            while(!vec.empty())
            {
                TreeNode* node=vec.front();
                tmp_res.push_back(node->val);
                if(node->left)
                    tmp_vec.push(node->left);
                if(node->right)
                    tmp_vec.push(node->right);
                vec.pop();
            }
            res.push_back(tmp_res);
            vec=tmp_vec;
        }
        reverse(res.begin(), res.end());
        return res;
    }
};