[JavaScript/section 5] 03 - 연속 부분 수열 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가지이다.

 

📝 풀이

🧑🏻‍💻 나의 풀이 방법

• lt(왼쪽을 담당할 포인터), rt(오른쪽을 담당할 포인터)를 인덱스 0으로 초기화한다.
• rt 포인터를 한칸씩 이동(증가)시키면서 포인터에 해당하는 값을 sum에 누적시킨다.
• 위 과정에서 sum이 m보다 크거나 같으면 다음 평가를 위해 while문을 돌면서 lt 포인터에 해당하는 값을 sum에서 빼주고 한칸 이동시켜야 한다.
• 이전에 sum과 m을 계속해서 비교하여 같다면 answer를 1 증가시킨다.

function solution(m, arr) {
  let answer = 0, sum = 0, lt = 0;

  for (let rt = 0; rt < arr.length; rt++) {
    sum += arr[rt];

    while (sum >= m) {
      if (sum === m) answer++;
      sum -= arr[lt++];
    }
  }

  return answer;
}

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

 

👨🏼‍🏫 강사님 풀이 방법

나의 풀이와 많은 부분이 비슷한 것을 확인할 수 있다. 단지 sum과 m을 언제 비교하느냐에 대한 차이다.

function solution(m, arr) {
  let answer = 0, lt = 0, sum = 0;

  for (let rt = 0; rt < arr.length; rt++) {
    sum += arr[rt];
    
    if (sum === m) answer++;
    
    while (sum >= m) {
      sum -= arr[lt++];
      if (sum === m) answer++;
    }
  }

  return answer;
}

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