====== 나머지 합 ====== ===== 풀이 ===== * 구간합 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]]도 거의 동일한 문제이다. ===== 코드 ===== """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() {{tag>BOJ ps:problems:boj:골드_3}}