Leetcode107:二叉树的层次遍历2

梦想不抛弃苦心追求的人,只要不停止追求,你们会沐浴在梦想的光辉之中。再美好的梦想与目标,再完美的计划和方案,如果不能尽快在行动中落实,最终只能是纸上谈兵,空想一番。只要瞄准了大方向,坚持不懈地做下去,才能够扫除挡在梦想前面的障碍,实现美好的人生蓝图。Leetcode107:二叉树的层次遍历2,希望对大家有帮助,欢迎收藏,转发!站点地址:www.bmabk.com,来源:原文

Leetcode107:二叉树的层次遍历2

题目:

在这里插入图片描述

思路:

按层次遍历树后将二维vector按逆序输出

迭代:

需使用queue模板类建立队列q,元素类型为指向树节点的指针。每次循环用一个一维vector存储一层树节点的值,通过出队列遍历并存储一层树节点,同时将它们的左右孩子入队。当q空时,遍历完毕退出。

先将树的根节点指针root入队,在进行遍历一层树节点前先创建一个一维vector用于存储一层树节点的值。计算当前q中的元素数,即为该层节点数,将其作为遍历循环的结束条件。在循环中将队头出队,将其指示节点的值
push到vector中,同时将左右孩子入队。循环上述过程。

退出循环后,将vector push到二维vector中。

当队列为空时,遍历结束,用STL中的reverse函数将二维vector变为逆序,二维vector的逆序即为vector的第一个行向量与最后一个行向量互换,第二个行向量与倒数第二个行向量互换,以此类推。

递归:

递归函数原型有三个参数:指向树中节点的指针p、层数level(从0开始算起)、用于保存结果的二维vector的引用ans

p为空为递归函数结束的条件,当ans.size()==level时,表明此时需新增行向量。接下来将p指示的节点的值push到ans[level]中,并分别对p的左右子树调用递归函数,它们的level参数为level+1。

在主函数中调用递归函数,p为树的根节点,level的值为0。得到二维vector后将其逆序即为所求。

代码:

迭代:
/**
 * 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<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>> ans;
        if( root==NULL )
            return ans;
        queue<TreeNode*> q;
        q.push( root );
        TreeNode* r;
        int n;
        int i;
        while( q.size()!=0 )
        {   
            vector<int> temp;
            n=q.size();
            for( i=0;i<n;i++ )
            {
                r=q.front();
                q.pop();
                temp.push_back( r->val );
                if( r->left!=NULL )
                    q.push( r->left );
                if( r->right!=NULL )
                    q.push( r->right );
            }
            ans.push_back( temp );
        }
        reverse( ans.begin(),ans.end() );
        return ans;
    }
};
递归:
/**
 * 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<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>> ans;
        leveltraverse( ans,0,root );
        reverse( ans.begin(),ans.end() );
        return ans;
    }
    void leveltraverse( vector<vector<int>> &ans,int level,TreeNode* p )
    {   
        if( p==NULL )
            return;
        if( ans.size()==level )
            ans.push_back( vector<int>() );
        ans[level].push_back( p->val );
        if( p->left!=NULL )
            leveltraverse( ans,level+1,p->left );
        if( p->right!=NULL )
            leveltraverse( ans,level+1,p->right );
    }
};

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/153870.html

(0)
飞熊的头像飞熊bm

相关推荐

发表回复

登录后才能评论
极客之音——专业性很强的中文编程技术网站,欢迎收藏到浏览器,订阅我们!