[백준] 2003 수들의 합2- Java

 

 

 

연속적인 합의 결과를 구하는 문제이니.. 딱 투 포인터로 풀면 되겠다고 생각된다.

 

 

처음 작성한 코드에서는 수열의 첫 번째 항이 구하려는 수와 같을 때를 잡아내지 못해서 헤멨다.

맞게 푼 것 같은데 틀리면 놓치기 쉬운 부분을 체크해보고, 극단적인 상황을 가정해서 테스트케이스를 유추해보자.

 

 

 

import java.util.*;
import java.io.*;
 
public class Main{			
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

		StringTokenizer st = new StringTokenizer(br.readLine());

		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());

		int[] arr = new int[N+1];

		st = new StringTokenizer(br.readLine());

		for(int i=1; i<N+1; i++){
			arr[i] = Integer.parseInt(st.nextToken());
		}
		

		int R = 1;
		int sum = arr[1];
		int ans = 0;

		if(sum == M){
			ans++;
		}

		for(int L = 1; L < N+1; L++){					
						
			while(R < N){
				
				if(sum + arr[R+1] > M){										
					break;
				}

				if(sum + arr[R+1] == M){			
					ans++;		
					break;
				}

				if(sum + arr[R+1] < M){
					R++;
					sum = sum + arr[R];					
				}
				
			}
			
			sum = sum - arr[L];

			if(sum == M){
				ans++;
			}
			

		}

		System.out.println(ans);
				 		
	}
}