<섹션3-Two pointers, Sliding window> 4. 연속 부분수열

조이·2022년 1월 6일
0

자바 알고리즘

목록 보기
28/41
post-thumbnail

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.*;

class Main {	
	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) {
		Main main = new Main();
		Scanner scan = new Scanner(System.in);
		
		int n=scan.nextInt();
		int m=scan.nextInt();
		int[] array=new int[n];
		for(int i=0;i<n;i++) array[i]=scan.nextInt();
		System.out.print(main.solution(n,m,array));
		}
}

<중요>

1) for문과 while문을 적절하게 잘 사용할 것!

profile
joy_study

0개의 댓글