LeetCode | Javascript

📍문제 설명📍

You are given an integer array nums consisting of n elements, and an integer k.

Find a contiguous subarray whose length is equal to k that has the maximum average value and return this value. Any answer with a calculation error less than 10-5 will be accepted.

📚입출력 예시📚

Example 1:

Input: nums = [1,12,-5,-6,50,3], k = 4
Output: 12.75000
Explanation: Maximum average is (12 - 5 - 6 + 50) / 4 = 51 / 4 = 12.75

Example 2:

Input: nums = [5], k = 1
Output: 5.00000

💯성공한 풀이💯

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var findMaxAverage = function(nums, k) {
    let max = -Infinity;
    let startWindow = 0;
    let sum = 0;
    
    for(let i = 0; i < nums.length; i++) {
        sum += nums[i];
        
        if(i - startWindow === k - 1) {
            let avg = sum / k;
            max = Math.max(max, avg);
            sum -= nums[startWindow];
            startWindow++;
        }
    }
    
    return max;
};

💡문제 접근 방식💡

슬라이딩 윈도우 문제이다.
첫 윈도우의 인덱스를 저장해둔다.
최대값을 찾아야 하므로, sum을 더하면서 윈도우의 끝 인덱스와 처음인덱스를 뺀 값이 k - 1 (윈도우 길이) 값과 같을때, 평균을내고 최대값을 구해준다.
윈도우를 옮길때는 이전 첫번째의 윈도우 값을 빼주면 된다.

처음 윈도우 [1, 12, -5, 6]
두번째 윈도우 [12, -5, -6, 50]
세번째 윈도우 [-5, -6, 50, 3]