[Algorithm]알고리즘 자바_27 연속 부분 수열

 

💬 문제 설명

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

8 6
1 2 1 3 1 1 1 2

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

 

3

 

 

​

💻 Solution.java

답안 1) 2중 for문으로 시간 복잡도가 좋지 못함

import java.util.*;

public class Main {
	public int solution(int n, int m, int[] arr) {
		int answer = 0, sum = 0;
		
		for(int i = 1; i <= n; i++) {
			for(int j = 0; j < n; j++) {
				if(j + i > n)
					break;
				for(int k = j; k < j + i; k++) {
					sum += arr[k];
				}
				System.out.println();
				if(sum == m)
					answer++;
				
				sum = 0;
			}
		}
		
        return answer;
	}

public static void main(String[] args){
		Main main  = new Main();
		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(main.solution(n, m, arr));
		
		sc.close();
		return ;
	}
}

답안 2) two pointer로 시간복잡도 O(n)으로 해결

import java.util.*;

public class Main {

	public int solution(int n, int m, int[] arr) {
		int answer = 0, sum = 0; 
		int 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){
		Main main  = new Main();
		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(main.solution(n, m, arr));
		
		sc.close();
		return ;
	}
}