IF - 연속 부분수열 2

Goody·2021년 4월 13일
0

알고리즘

목록 보기
86/122

문제

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가지입니다.


예시

MarrOutput
5[1,3,1,2,3]10

풀이 및 회고

  • 연속 부분수열 1 문제 하고 비슷하게, 루프를 두 번 돌면서 각 원소마다 수열을 검사해준다.
  • 수열 내 원소의 개수는 제한하지 않으므로, 원소가 하나일 때도 m보다 작다면 count를 1 늘려준다.
  • 수열 검사중인 원소와 뒤의 원소와의 합이 m보다 작으면 count를 1 늘려주고, 합을 다음 루프로 넘긴다.
  • 합이m 보다 커지면 두 번째 루프를 중단시킨다.

코드

const solution =(m, sequence) => {

    let count = 0;
    
    for(let i = 0; i < sequence.length; i++) {
        let temp = 0;
        if(sequence[i] < m) {
            temp += sequence[i];
            count++;
        }
        for(let j = i+1; j < sequence.length; j++) {
            if(temp + sequence[j] <= m) {
                temp += sequence[j];
                count++;
            }else {
                break;
            }
        }
    }
    
    return count;
}

0개의 댓글