Moving Average from Data Stream

lc 346 Moving Average from Data Stream

Given a stream of integers and a window size, calculate the moving average of all integers in the sliding window.

For example,

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

public class MovingAverage { LinkedList queue; int size; int sum;

/** Initialize your data structure here. */
public MovingAverage(int size) {
    this.queue = new LinkedList<Integer>();
    this.size = size;
}

public double next(int val) {
    if(queue.size()<this.size){
        queue.offer(val);
        if(queue.size() == 1) sum = val;
        else{
            sum += val;
        }
        double avg = (double)sum/queue.size();

        return avg;
    }else{
        int head = queue.poll();
        sum -= head;
        queue.offer(val);
        sum += val;
        double avg = (double)sum/queue.size();
        return avg;
    }
}

}