【leetcode】102 103 637 典型BFS

【102.Binary Tree Level Order Traversal】
这里写图片描述
「思路」
运用广度优先算法(BFS)的思想。用一个队列q,在处理并pop上一层的数据的同时push近其下一层左右子树,pop完上一层时,下一层的数据也刚好完全push入队列,以此类推完成操作。
「运行结果」
这里写图片描述
「代码」

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>>ans;
        if(!root)return ans;
        queue <TreeNode*> q;
        q.push(root);
        while(!q.empty())
        {
            int num=q.size();
            vector<int> group;
            for(int i=0;i<num;i++)
            {
                TreeNode*t=q.front();
                group.push_back(t->val);
                q.pop();
                if(t->left) q.push(t->left);
                if(t->right)q.push(t->right);
            }
            ans.push_back(group);
        }
        return ans;
    }
};

【637.Average of Levels in Binary Tree】
这里写图片描述
「思路」
同样是典型的BFS题,与上题思路大致相同,就是多加了一个总和在循环里。
「运行结果」
这里写图片描述
「代码」

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
         vector<double>ans;
         queue<TreeNode*>q;
        q.push(root);
        while(!q.empty())
        {
            int num=q.size();
            double sum=0;
            for(int i=0;i<num;i++)
            {
                TreeNode*t=q.front();
                q.pop();
                if(t->left) q.push(t->left);
                if(t->right)q.push(t->right);
                sum+=t->val;
            }
            ans.push_back(sum/num);
        }
        return ans;
    }
};

【103. Binary Tree Zigzag Level Order Traversal】
这里写图片描述
「思路」
在102的基础上,最后加一个用于反转代码,格一行将数据反转一次。
「运行结果」
这里写图片描述
「代码」

class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
         vector<vector<int>>ans;
        if(!root)return ans;
        queue <TreeNode*> q;
        q.push(root);
        //int z=0;
        while(!q.empty())
        {
            int num=q.size();
            vector<int> group;
            for(int i=0;i<num;i++)
            {
                TreeNode*t=q.front();
                group.push_back(t->val);
                q.pop();
                if(t->left) q.push(t->left);
                if(t->right)q.push(t->right);

            }
            ans.push_back(group);
        }
        for(int i=1;i<ans.size();i=i+2)
        {
            reverse(ans[i].begin(),ans[i].end());
        }
        return ans;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值