二叉树的层平均值

我怀疑我今天要把二叉树的题目全做完hhh
还是之前那个层的基础上改动了一点
二叉树层的平均值

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