ps:problems:boj:16993
                연속합과 쿼리
| ps | |
|---|---|
| 링크 | acmicpc.net/… | 
| 출처 | BOJ | 
| 문제 번호 | 16933 | 
| 문제명 | 연속합과 쿼리 | 
| 레벨 | 플래티넘 2 | 
| 분류 | 
 구간 쿼리  | 
	
| 시간복잡도 | O(n + mlogn) | 
| 인풋사이즈 | n<=100,000, m<=100,000 | 
| 사용한 언어 | Python | 
| 제출기록 | 62240KB / 2840ms | 
| 최고기록 | 2840ms | 
| 해결날짜 | 2021/03/21 | 
풀이
- 구간내 최대 연속부분합 쿼리, 일명 '금광세그' 쿼리 문제.
 - 세그먼트 트리의 각 노드에 저장할 값과, 그 값을 합치는 방법이 조금 트리키하긴 하지만, 구현상으로는 Teferi library의 bottom-up 세그먼트 트리 구현체를 그대로 사용하는데에 아무 문제가 없다.
 - 구간값을 합치는 것이 O(1)에 되므로, 각 쿼리는 O(logn)에 처리 가능하다. 그러면 트리 구축에 O(n)이 걸리고, m개의 쿼리를 각각 O(logn)에 처리하므로 총 시간 복잡도는 O(n+logm),
 
코드
"""Solution code for "BOJ 16993. 연속합과 쿼리".
- Problem link: https://www.acmicpc.net/problem/16993
- Solution link: http://www.teferi.net/ps/problems/boj/16993
"""
import sys
from teflib import segmenttree
def merge(l, r):
    l_lmax, l_rmax, l_max, l_all = l
    r_lmax, r_rmax, r_max, r_all = r
    return (max(l_lmax, l_all + r_lmax), max(r_rmax, l_rmax + r_all),
            max(l_max, r_max, l_rmax + r_lmax), l_all + r_all)
def main():
    N = int(sys.stdin.readline())
    A = [int(x) for x in sys.stdin.readline().split()]
    M = int(sys.stdin.readline())
    segtree = segmenttree.SegmentTree(((x, x, x, x) for x in A), merge)
    for _ in range(M):
        i, j = [int(x) for x in sys.stdin.readline().split()]
        print(segtree.query(i - 1, j)[2])
if __name__ == '__main__':
    main()
- Dependency: teflib.segmenttree.SegmentTree
 
ps/problems/boj/16993.txt · 마지막으로 수정됨: 2021/04/30 15:16 저자 teferi
                
                
토론