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()
ps/problems/boj/23327.txt · 마지막으로 수정됨: 2023/07/24 16:22 저자 teferi
토론