leetcode 107: 二叉树的层次遍历 II

对于二叉树来说,递归是最好用的方式

void levelPush(TreeNode*l1,int level,std::vector<std::vector<int>>&a){
    if(l1==NULL)return;
    if(level==a.size()){
        std::vector<int> b;
        a.push_back(b);
    }
    a[level].push_back(l1->val);
    levelPush(l1->left,level+1,a);
    levelPush(l1->right,level+1,a);
}


std::vector<std::vector<int>> levelOrderBottom(TreeNode*root){
    std::vector<std::vector<int>> a;
    levelPush(root,0,a);
    return std::vector<std::vector<int>>(a.rbegin(),a.rend());
}

方式二:队列

std::vector<std::vector<int>> levelOrderBottom(TreeNode *root){
    std::vector<std::vector<int>> a;
    if(root==NULL)return a;
    std::deque<TreeNode*> q;
    std::vector<int>v;
    q.push_back(root);
    while(!q.empty()){
        int c=q.size();
        for(int i=0;i<c;i++){
            TreeNode *l=q.front();
            q.pop_front();
            v.push_back(l->val);
            if(l->left!=NULL)
                q.push_back(l->left);
            if(l->right!=NULL)
                q.push_back(l->right);
        }
        a.insert(a.begin(),v);
        v.clear();
    }
    return a;
}

猜你喜欢

转载自blog.csdn.net/u013263891/article/details/82876110