[백준] P2003 수들의 합2

https://www.acmicpc.net/problem/2003

2003번: 수들의 합 2

 

🖊️문제

N개의 수로 된 수열 A[1], A[2], …, A[N] 이 있다. 이 수열의 i번째 수부터 j번째 수까지의 합 A[i] + A[i+1] + … + A[j-1] + A[j]가 M이 되는 경우의 수를 구하는 프로그램을 작성하시오.

 

🖊️문제 풀이

나는 이 문제를 투포인터를 이용해서 풀었다.

투포인터 관련 설명은 이 링크를 참고하시면 좋을 듯 해서 어떤 분이 올리신 깃허브 글 링크 달아두기루

https://github.com/WooVictory/Ready-For-Tech-Interview/blob/master/Algorithm/%ED%88%AC%ED%8F%AC%EC%9D%B8%ED%84%B0%20%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98.md

 

투포인터 문제 중에서도 조금 쉬운 문제에 속하는 듯?

 

🖊️코드

package 투포인터;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class P2003 {

  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];

    st = new StringTokenizer(br.readLine());
    for (int i = 0; i < n; i++) {
      arr[i] = Integer.parseInt(st.nextToken());
    }

    int cnt=0;
    int sum = 0;
    int start = 0;
    int end = 0;

    while (true){
      if(end == n){
        break;
      }else if(sum < m){
        sum+=arr[end++];
      }else{
        sum-=arr[start++];
      }

      if(sum==m){
        cnt++;
      }
    }

    System.out.println(cnt);
  }

}