사용자 도구

사이트 도구


ps:problems:boj:11969

Breed Counting

ps
링크acmicpc.net/…
출처BOJ
문제 번호11969
문제명Breed Counting
레벨실버 3
분류

누적합

시간복잡도O(n+q)
인풋사이즈n<=100,000, q<=100,000
사용한 언어Python
제출기록37820KB / 520ms
최고기록424ms
해결날짜2022/05/29
태그

[라이] 구간합 배열

풀이

  • 소의 마리수의 누적합 배열을 3종류의 id에 대해 각각 만들어주면 된다. 쿼리가 주어지면, 각 누적합 배열에서 부분합을 각각 계산해주면 된다.
  • 누적합 배열을 만드는데 O(n), q개의 쿼리를 각각 O(1)로 처리하는 데에 O(q). 총 O(n+q).

코드

"""Solution code for "BOJ 11969. Breed Counting".

- Problem link: https://www.acmicpc.net/problem/11969
- Solution link: http://www.teferi.net/ps/problems/boj/11969

Tags: [Prefix sum]
"""

import sys


def main():
    N, Q = [int(x) for x in sys.stdin.readline().split()]  # pylint: disable=unused-variable
    ids = [sys.stdin.readline().rstrip() for _ in range(N)]

    prefix_sums = [[v := 0] + [v := (v + 1 if x == '1' else v) for x in ids],
                   [v := 0] + [v := (v + 1 if x == '2' else v) for x in ids],
                   [v := 0] + [v := (v + 1 if x == '3' else v) for x in ids]]
    for _ in range(Q):
        a, b = [int(x) for x in sys.stdin.readline().split()]
        print(*(psum[b] - psum[a - 1] for psum in prefix_sums))


if __name__ == '__main__':
    main()

토론

댓글을 입력하세요:
W F A J U
 
ps/problems/boj/11969.txt · 마지막으로 수정됨: 2022/05/31 14:24 저자 teferi