题目描述:
给定一个整数数据流和一个窗口大小,根据该滑动窗口的大小,计算其所有整数的移动平均值。
示例:
MovingAverage m = new MovingAverage(3);
m.next(1) = 1
m.next(10) = (1 + 10) / 2
m.next(3) = (1 + 10 + 3) / 3
m.next(5) = (10 + 3 + 5) / 3
方法1:
主要思路:
(1)直观的想,使用队列维护输入的元素,使用m_size 标识元素数量的限制,使用 m_sum 统计当前元素的和,最后直接返回当前元素的和除以当前元素的数量即可;
class MovingAverage {
public:
queue<int> m_q;//维护元素
int m_size;//标识限制数量
int m_sum;//保持当前元素的和
/** Initialize your data structure here. */
MovingAverage(int size):m_size(size),m_sum(0) {//初始化元素初始值
}
double next(int val) {
//更新元素
if(m_q.size()<m_size){
m_sum+=val;
m_q.push(val);
}
else{
m_sum-=m_q.front();
m_q.pop();
m_sum+=val;
m_q.push(val);
}
return (double(m_sum))/m_q.size();
}
};
/**
* Your MovingAverage object will be instantiated and called as such:
* MovingAverage* obj = new MovingAverage(size);
* double param_1 = obj->next(val);
*/