【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;
    }
};