[인프런] Node.js / 섹션5-효율성(투포인터 알고리즘) / 3. 연속 부분수열 1

N개의 수로 이루어진 수열, 이 수열에서 연속부분수열의 합이 특정숫자 M이 되는 경우가 몇 번 있는지 구하기.

예를들어 N=8, M=6이고 수열이 1 2 1 3 1 1 1 2 라면,

합이 6이 되는 연속부분수열은 {2, 1, 3}, {1, 3, 1, 1}, {3, 1, 1, 1}로 총 3가지

 

풀이

function solution(m, arr) {
  let p1 = 0;
  let p2 = 0;
  let count = 0;
  let subtotal = arr[p2];

  while (p1 < arr.length && p2 < arr.length) {
    if (subtotal < m) {
      subtotal += arr[++p2];
    } else {
      subtotal -= arr[p1++];
    }
    if (subtotal === m) {
      count++;
    }
  }

  return count;
}

let a = [1, 2, 1, 3, 1, 1, 1, 2];
console.log(solution(8, a));

투포인터가 아직 익숙치 않아서 시각적으로 확인하고자 그림을 그려보면서 접근했다.