사용자 도구

사이트 도구


ps:problems:boj:10986

나머지 합

ps
링크acmicpc.net/…
출처BOJ
문제 번호10986
문제명나머지 합
레벨골드 3
분류

누적합

시간복잡도O(n)
인풋사이즈n<=10^6
사용한 언어Python
제출기록148896KB / 600ms
최고기록484ms
해결날짜2022/05/31
태그

[단계]누적 합, [라이]구간합 배열

풀이

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

토론

댓글을 입력하세요:
D I E F D
 
ps/problems/boj/10986.txt · 마지막으로 수정됨: 2022/11/18 01:57 저자 teferi