사용자 도구

사이트 도구


ps:problems:boj:23327

리그전 오브 레전드

ps
링크acmicpc.net/…
출처BOJ
문제 번호23327
문제명리그전 오브 레전드
레벨골드 3
분류

누적합

시간복잡도O(n+q)
인풋사이즈n<=100,000, q<=200,000
사용한 언어Python 3.11
제출기록45644KB / 528ms
최고기록436ms
해결날짜2023/07/24

풀이

  • $ (\sum_i{a_i})^2 = {\sum_i{a_i^2}} + 2\sum_i\sum_{j>i}{a_ia_j} $ 라는 것만 떠올리면 된다.
  • a_i의 누적합과 a_i^2의 누적합을 구해두면, 주어진 쿼리들을 각 O(1)에 처리할수 있다
  • 시간복잡도는 O(n+q)

코드

"""Solution code for "BOJ 23327. 리그전 오브 레전드".

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

Tags: [prefix sum]
"""

import sys


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

    prefix_sum = [v := 0] + [v := v + x for x in a]
    prefix_sq_sum = [v := 0] + [v := v + x * x for x in a]

    for _ in range(Q):
        l, r = [int(x) - 1 for x in sys.stdin.readline().split()]

        lr_sum = prefix_sum[r + 1] - prefix_sum[l]
        lr_sq_sum = prefix_sq_sum[r + 1] - prefix_sq_sum[l]
        answer = (lr_sum * lr_sum - lr_sq_sum) // 2

        print(answer)


if __name__ == '__main__':
    main()

토론

댓글을 입력하세요:
F Z V B R
 
ps/problems/boj/23327.txt · 마지막으로 수정됨: 2023/07/24 16:22 저자 teferi