연속 부분수열2

문제

N개의 수로 이루어진 수열이 주어집니다.
이 수열에서 연속부분수열의 합이 특정숫자 M이하가 되는 경우가 몇 번 있는지 구하는 프로그
램을 작성하세요.
만약 N=5, M=5이고 수열이 다음과 같다면
1 3 1 2 3
합이 5이하가 되는 연속부분수열은 {1}, {3}, {1}, {2}, {3}, {1, 3}, {3, 1}, {1, 2}, {2, 3},
{1, 3, 1}로 총 10가지입니다.

제한사항

입출력 예

풀이

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) {
      sum -= arr[lt++];
    }
    answer += rt - lt + 1;
  }
  return answer;
}

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

• 수학적인 원리가 조금 추가된 투포인터 문제이다.
• rt,lt는 0부터 시작이다. 합이 m이하가 되는 부분수열을 다 찾는 것인데, sum > m을 넘지 않는 선에서 부분수열의 개수는 마지막 인덱스 번호 - 시작 인덱스 번호 + 1 이다.
  (개수를 구하기 때문에 +1)