연속 부분수열2(효율성-투 포인터 알고리즘)

문제(출처 : 인프런 자바스크립트 알고리즘 문제풀이 강의, 정보올림피아드)

 

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

▣ 입력설명
첫째 줄에 N(1≤N≤100,000), M(1≤M≤100,000,000)이 주어진다. 수열의 원소값은 1,000을 넘지 않는 자연수이다.

▣ 출력설명
첫째 줄에 경우의 수를 출력한다.

 

▣ 입력예제 1

5 5

1 3 1 2 3

 

▣ 출력예제 1

10

Tip

 

문제풀이

//강의 듣고 내가 다시 작성한 답
function solution(arr,num){
    let answer = 0;
    let sum = 0;
    let lp = 0;

    for(let rp = 0 ; rp < arr.length ; rp++){
        sum += arr[rp];
        while(sum > num){
            sum -= arr[lp++];
        }
        answer += (rp - lp + 1);
    }
    return answer;
}

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