[BJ] 2003번: 수들의 합2 (Java)

💡 문제

2003번: 수들의 합 2

문제

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

입력

첫째 줄에 N(1 ≤ N ≤ 10,000), M(1 ≤ M ≤ 300,000,000)이 주어진다. 다음 줄에는 A[1], A[2], …, A[N]이 공백으로 분리되어 주어진다. 각각의 A[x]는 30,000을 넘지 않는 자연수이다.

출력

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

예제 입력 1

4 2
1 1 1 1

예제 출력 1

3

---

💡 접근 방법

---

🤦‍♀️ My Solution

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

public class Main {
    static int N, M, total;
    static int arr[];

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

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

        total = 0;
        sum();
        System.out.println(total);
    }

    static public void sum() {

        boolean t = true;
        while (t) {
            for (int i = 0; i < N; i++) {
                if (arr[i] == M) {
                    total++;
                    continue;
                } else {
                    int sum2 = 0;
                    sum2 = arr[i];
                    for (int j = i + 1; j < N; j++) {

                        sum2 += arr[j];

                        if (sum2 > M) {
                            sum2 = 0;
                            break;
                        }
                        if (sum2 == M) {
                            total++;
                            sum2 = 0;
                            break;
                        }
                    }
                }
            }
            t = false;
        }

    }
}