백준_10986번_나머지합구하기

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

10986번: 나머지 합

import java.util.*;

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		
		Scanner sc = new Scanner(System.in);
		
		int N = sc.nextInt(); // 입력받을 숫자갯수 입력 
		int M = sc.nextInt(); // 나눌려고 하는 수 입력
		
		long sum [] = new long [N]; //숫자 입력 할 배열 
		long count [] = new long [M]; //합배열 
		
		long result = 0; //초기화 
		
		sum[0] = sc.nextInt();
		
		for(int i = 1; i < N; i++) {
			
			sum[i] = sum[i-1] + sc.nextInt();
		}
		
		for(int i = 0; i < N; i++) {
			
			int remain = (int)(sum[i] % M);
			if(remain == 0) {
				result++;
			}
			
			count[remain]++;
		}
		
		for(int i=0; i<M; i++) {
			
			if(count[i] > 1) {
				long cnt = count[i];
				result = result + (cnt * (cnt - 1) / 2);
			}
		}
		
		System.out.println(result);
		
	}

}