Sword finger offer II 041. Average value of sliding window

Original link ： The finger of the sword Offer II 041. Average value of sliding window

 

solution

class MovingAverage {
public:
    /** Initialize your data structure here. */
    int wsize;  // Window size 
    vector<int> window;
    int kth = 0;
    int flag = 0;
    MovingAverage(int size) {
        wsize = size;
        window = vector<int> (size);
    }
    
    double next(int val) {
        double res = 0; // Define the return value 
        kth %= wsize;
        window[kth] = val;
        kth++;
        flag++;
        for(auto &x : window) res += x;
        return flag >= wsize? res / wsize : res / kth;
    }
};

/**
 * Your MovingAverage object will be instantiated and called as such:
 * MovingAverage* obj = new MovingAverage(size);
 * double param_1 = obj->next(val);
 */