给出一串整数流和窗口大小,计算滑动窗口中所有整数的平均值。
样例
样例1 :
MovingAverage m = new MovingAverage(3);
m.next(1) = 1 // 返回 1.00000
m.next(10) = (1 + 10) / 2 // 返回 5.50000
m.next(3) = (1 + 10 + 3) / 3 // 返回 4.66667
m.next(5) = (10 + 3 + 5) / 3 // 返回 6.00000
思路:利用queue,每次往后入列,前面出列的特点,将每个数存入,再利用sum储存每一次加入的数,若不超过size,则sum除以queue的len,若超过,则sum需减去queue中最前的一个数,再除以size。
class MovingAverage {
public:
/*
* @param size: An integer
*/
MovingAverage(int size) {
// do intialization if necessary
m_size=size;
}
/*
* @param val: An integer
* @return:
*/
double next(int val) {
// write your code here
result.push(val);
int len=result.size();
if(len<=m_size)
{
sum+=val;
return sum/len;
}
else
{
sum-=result.front();
sum+=result.back();
result.pop();
return sum/m_size;
}
}
private:
int m_size;
queue<int> result;
double sum;
};