637. 二叉树的层平均值 递归写法 c++

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.

示例 1:

输入:
3
/
9 20
/
15 7
输出: [3, 14.5, 11]
解释:
第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].

class Solution
{
public:
	vector<pair<double,int>> res;//层的累加和，层节点个数
	void Mid_Search_BT(TreeNode* root, int n)//中序遍历
	{
		if (root == NULL)
			return;
		else
		{
			if (n >= res.size())
			{
				res.push_back(make_pair(0.0,0));
			}
			Mid_Search_BT(root->left, n + 1);
			res[n].first += root->val;
			res[n].second++;
			Mid_Search_BT(root->right, n + 1);
		}
	}
	vector<double> averageOfLevels(TreeNode* root)
	{
		vector <double>result;
		Mid_Search_BT(root, 0);

		for (int i = 0; i < res.size(); i++)
		{
			result.push_back(res[i].first / (res[i].second*1.0));
		}
		return result;
	}
};