연속 부분 수열

22 Apr 2023 in Algorithm on Java, Coding, Test

문제

설명

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

입력

첫째 줄에 N(1≤N≤100,000), M(1≤M≤100,000,000)이 주어진다.

수열의 원소값은 1,000을 넘지 않는 자연수이다.

출력

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

예시 입력 1

8 6
1 2 1 3 1 1 1 2

예시 출력 1

3

해결방법

• 반복 부터 돌릴 생각 하지 말고, 시간 복잡도를 줄이기 위한 방법을 생각할 것.

• 포인터를 옮겨가며 비교.

  

코드

import java.util.*;

class Main {
    public int solution(int n, int m, int[] arr) {
        int answer = 0, sum = 0, lt = 0;
        // 연속 부분 수열의 시작 (슬라이딩 윈도우) 을 가리키는
        // 포인터가 2개 (왼,오) 존재한다.
        for (int rt = 0; rt < n; rt++) {
            // 오른쪽 포인터부터 시작하여 합계에 더한 후, 비교한다.
            // 오른쪽 부터 계속 원소를 더하다보면 합이 6이 되는 경우가 나오거나 큰 경우가 발생한다.
            // 경우가 발생하면 증가 시키고, 원소를 뺐을 때 상황을 체크한다.
            // 답을 증가 시켰으면 다시 더하기 시작하여 체크한다.
            sum += arr[rt];
            // 합계를 비교한 후, 답을 증가시킨다.
            if (sum == m) answer++;
            // 만약, 합계보다 크거나 같다면 왼쪽 포인터 부터 합에서 제외시킨다.
            while (sum >= m) {
                sum -= arr[lt++];
                if (sum == m) answer++;
            }
        }
        return answer;
    }

    public static void main(String[] args) {
        Main.Main T = new Main.Main();
        Scanner kb = new Scanner(System.in);
        int n = kb.nextInt();
        int m = kb.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = kb.nextInt();
        }
        System.out.print(T.solution(n, m, arr));
    }
}