나머지_합


title: 2026-02-17 author: 강병호 (이름) date: 2026-02-17 (날짜) category: TIL/강병호/2026/02 (파일 경로 : TIL/{이름}/{연}/{월}) layout: post (자유) —

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

        // 나머지 0 => 무조건
        // 나머지 0, 1, 2 속에서 3C2 -> 0이되는 개수

        int arr[] = new int[N];
        long[] cnt = new long[M]; // 0, 1, 2
        long sum = 0;

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

        for (int i = 0; i < N; i++) {
            sum = (sum + arr[i]) % M;
            int index = (int) sum;
            cnt[index]++;
        }

        long ans = cnt[0];

        for (int i = 0; i < M; i++) {
            ans += ((cnt[i] * (cnt[i]-1)) / 2);
        }

        System.out.println(ans);
    }
}

results matching ""

    No results matching ""