[Inflearn] 자바 알고리즘 문제풀이 #03-04 4. 연속 부분 수열

문제

설명

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을 넘지 않는 자연수이다.

 

출력

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

 

 

 

내 풀이

import java.util.Scanner;

public class Main3_4 {
    public int Solution(int n, int m, int[] arr) {
        int sum=0, p1=0, p2=0, answer=0;
        while(sum<m && p2<n) {
            sum += arr[p2++];
            if(sum==m) answer++;

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

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

03-03 문제에서 배웠던 Sliding window을 응용해봤다.

two pointer를 사용해서 p2는 6이 넘어갈 때까지 배열을 더해주는데,

6이 딱 맞아 떨어지게되면 정답+1 해주고 6보다 작아질 때까지 p1을 이용해 맨 왼쪽 값부터 값을 빼주고

6보다 큰 숫자가 되면 6보다 작아질 때까지 p1을 이용해 값을 빼주는데 빼다가 6이 되면 정답+1 해주는 것이다.

 

 

 

선생님 풀이

import java.util.Scanner;

public class Main3_4 {
    public int Solution(int n, int m, int[] arr) {
        int answer=0, sum=0, lt=0;
        for (int rt = 0; rt < n; rt++) {
            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) {
        Main3_4 T = new Main3_4();
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        System.out.println(T.Solution(n, m, arr));
    }
}

선생님이랑 비슷하게 풀었지롱~~~ 개뿌듯~~~

뭔가 그동안 time limit를 많이 만나서 for문을 쓰면 안될 것 같은 강박(?)이 생긴 것 같다ㅋㅋㅋ

그래서 나는 while문으로만 했는데 하나 정도는 괜찮군 ㅎㅎ,,,

 

 

 

결과