BOJ 10986 나머지 합

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

문제

수 N개 $A_1, A_2, …, A_N$이 주어진다. 이때, 연속된 부분 구간의 합이 $M$으로 나누어 떨어지는 구간의 개수를 구하는 프로그램을 작성하시오.

즉, $A_i + … + A_j (i ≤ j)$ 의 합이 $M$으로 나누어 떨어지는 $(i, j)$ 쌍의 개수를 구해야 한다.

---

입력

첫째 줄에 $N$과 $M$이 주어진다. $(1 ≤ N ≤ 10^6, 2 ≤ M ≤ 10^3)$

둘째 줄에 $N$개의 수 $A_1, A_2, …, A_N$이 주어진다. $(0 ≤ A_i ≤ 10^9)$

---

출력

첫째 줄에 연속된 부분 구간의 합이 M으로 나누어 떨어지는 구간의 개수를 출력한다.

---

예제 입력 1

5 3
1 2 3 1 2

예제 출력 1

7

---

코드

아이디어가 잘 떠오르지 않아서 다른 포스트를 참고했다.

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

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());

        st = new StringTokenizer(br.readLine());
        int[] restCount = new int[1000];
        int sum = 0;
        long answer = 0;
        for (int i = 0; i < N; i++) {
            int input = Integer.parseInt(st.nextToken());
            sum += input;
            sum %= M;
            if(sum == 0) {
                answer++;
            }

            answer += restCount[sum];
            restCount[sum]++;
        }

        System.out.println(answer);
    }
}