테페리넷

이 문서는 읽기 전용입니다. 원본을 볼 수는 있지만 바꿀 수는 없습니다. 문제가 있다고 생각하면 관리자에게 문의하세요.

====== 나머지 합 ======

===== 풀이 =====
  * 구간합 A[i..j] 가 M의 배수가 된다는 것은 A[0..j]를 M으로 나눈 나머지와 A[0..i-1]를 M으로 나눈 나머지가 같다는 뜻이다. 결국 누적합을 구하듯이 A[0..i]%M 을 저장한 배열을 만들고, 거기에서 값이 같은 쌍의 갯수를 찾으면 된다.
  * 누적합 배열을 만드는 것은 O(N). 값이 같은 쌍의 갯수를 찾는것은 그냥 배열을 한번 이터레이션하면서 구해도 되고, 각 값별로 카운트를 세어놓고서 계산해도 된다. 어떻게 하든 O(N)으로 충분. 그래서 총 시간복잡도도 O(N)
  * [[ps:problems:boj:3673]]도 거의 동일한 문제이다.

===== 코드 =====
<dkpr py>
"""Solution code for "BOJ 10986. 나머지 합".

- Problem link: https://www.acmicpc.net/problem/10986
- Solution link: http://www.teferi.net/ps/problems/boj/10986

Tags: [Prefix sum]
"""

import collections


def main():
    N, M = [int(x) for x in input().split()]  # pylint: disable=unused-variable
    A = [int(x) for x in input().split()]

    prefix_sum = [ps := 0] + [ps := (ps + x) % M for x in A]
    counter = collections.Counter(prefix_sum)
    print(sum(x * (x - 1) // 2 for x in counter.values()))


if __name__ == '__main__':
    main()
</dkpr>

{{tag>BOJ ps:problems:boj:골드_3}}