leetcode 637、429 学习笔记（模板题，二叉树层序遍历）

637.二叉树的层平均值

问题描述：传送门

思路：

1、统计每层值的总和，以及节点个数。

代码：

	class Solution {
	public:
	    vector<double> averageOfLevels(TreeNode* root) {
	        queue<TreeNode*> que;
	        if (root != NULL) que.push(root);
	        vector<double> result;
	        while (!que.empty()) {
	            int size = que.size();
	            double sum = 0; // 统计每一层的和
	            for (int i = 0; i < size; i++) { 
	                TreeNode* node = que.front();
	                que.pop();
	                sum += node->val;
	                if (node->left) que.push(node->left);
	                if (node->right) que.push(node->right);
	            }
	            result.push_back(sum / size); // 将每一层均值放进结果集
	        }
	        return result;
	    }
	};

429.N叉树的层序遍历

问题描述：传送门

思路：

1、这道题依旧是模板题，只不过一个节点有多个孩子了

代码：

	class Solution {
	public:
	    vector<double> averageOfLevels(TreeNode* root) {
	        queue<TreeNode*> que;
	        if (root != NULL) que.push(root);
	        vector<double> result;
	        while (!que.empty()) {
	            int size = que.size();
	            double sum = 0; // 统计每一层的和
	            for (int i = 0; i < size; i++) { 
	                TreeNode* node = que.front();
	                que.pop();
	                sum += node->val;
	                if (node->left) que.push(node->left);
	                if (node->right) que.push(node->right);
	            }
	            result.push_back(sum / size); // 将每一层均值放进结果集
	        }
	        return result;
	    }
	};

今天又学到新的知识，开心哦！