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