Sliding Window, Maximum Average Subarray 문제 (leet code)

문제 링크

https://leetcode.com/problems/maximum-average-subarray-i

해결 방법

• subarray 라는 용어를 보고 sliding window 알고리즘으로 접근해본다.
• 길이 k 가 고정되어 있으므로, 초기에 윈도우 사이즈만큼 더하며 초기화를 해준다.
• 초기화 후에 sliding window 를 우측으로 한칸씩 옮기며 모든 window 를 점검하면 끝난다.

소스코드

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var findMaxAverage = function (nums, k) {
  // subarray -> sliding window
  // init variables to solve sliding windows.
  let l = 0;
  let r = 0;
  let c = 0;

  // 1. sum till k - 1 first and save average to ans
  for (; r < k; r++) {
    c += nums[r];
  }

  let ans = c / k;

  // while r < nums.length
  while (r < nums.length) {
    // 2. subtract nums[l], l++
    c -= nums[l++];
    // 3. add nums[r], r++
    c += nums[r++];
    // 4. update ans with max function
    ans = Math.max(ans, c / k);
  }

  return ans;
};