ps:problems:boj:24838
                배열 구간합 놀이
| ps | |
|---|---|
| 링크 | acmicpc.net/… | 
| 출처 | BOJ | 
| 문제 번호 | 24838 | 
| 문제명 | 배열 구간합 놀이 | 
| 레벨 | 골드 4 | 
| 분류 | 
 구간합  | 
	
| 시간복잡도 | O(m+nlogn) | 
| 인풋사이즈 | m<=200,000, n<=50,000 | 
| 사용한 언어 | Python 3.11 | 
| 제출기록 | 46992KB / 2444ms | 
| 최고기록 | 2444ms | 
| 해결날짜 | 2023/08/01 | 
풀이
- 각 인덱스가 몇개의 구간에 포함되는지를 구하고 나면, 가장 많은 구간에 포함되는 인덱스에 가장 큰 값을 배치하는 그리디한 방법으로 총합을 가장 크게 만들수 있다,
 - 각 인덱스가 몇개의 구간에 포함되는지는, 누적합과 계차수열 트릭을 이용해서 O(m+n)에 쉽게 구할수 있다.
 - 추가적으로 구해야 하는 것은, 최댓값을 달성하는 배치 방법의 갯수이다. 포함되는 구간의 갯수가 같은 인덱스끼리는 서로 교환해도 총합이 동일하다. 따라서 포함되는 구간의 개수가 같은 인덱스들을 그룹으로 묶어준 다음에, 각 그룹에 대해서 그룹크기의 팩토리얼을 계산해서 곱해주면, 총합을 동일하게 유지하는 배치방법의 개수가 구해진다
 - 총 시간복잡도는 O(m+n) + O(nlogn) + O(n) = O(m+nlogn)
 
코드
"""Solution code for "BOJ 24838. 배열 구간합 놀이".
- Problem link: https://www.acmicpc.net/problem/24838
- Solution link: http://www.teferi.net/ps/problems/boj/24838
"""
import collections
import math
import sys
MOD = 10**9 + 7
def main():
    T = int(sys.stdin.readline())
    for _ in range(T):
        n, m = [int(x) for x in sys.stdin.readline().split()]
        count_delta = [0] * (n + 1)
        A = [int(x) for x in sys.stdin.readline().split()]
        for _ in range(m):
            x, y = [int(x) for x in sys.stdin.readline().split()]
            count_delta[x - 1] += 1
            count_delta[y] -= 1
        v = 0
        add_count = [v := v + x for x in count_delta]
        add_count.pop()
        max_sum = sum(x * y for x, y in zip(sorted(A), sorted(add_count)))
        counter = collections.Counter(add_count)
        arrangement_count = (
            math.prod(math.factorial(v) % MOD for v in counter.values()) % MOD
        )
        print(max_sum, arrangement_count)
if __name__ == '__main__':
    main()
ps/problems/boj/24838.txt · 마지막으로 수정됨: 2023/08/01 07:50 저자 teferi
                
                
토론