| ps | |
|---|---|
| 링크 | acmicpc.net/… |
| 출처 | BOJ |
| 문제 번호 | 10986 |
| 문제명 | 나머지 합 |
| 레벨 | 골드 3 |
| 분류 |
누적합 |
| 시간복잡도 | O(n) |
| 인풋사이즈 | n<=10^6 |
| 사용한 언어 | Python |
| 제출기록 | 148896KB / 600ms |
| 최고기록 | 484ms |
| 해결날짜 | 2022/05/31 |
| 태그 | |
"""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()