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()
ps/problems/boj/10986.txt · 마지막으로 수정됨: 2022/11/18 01:57 저자 teferi
토론