ps:problems:boj:1492
                합
| ps | |
|---|---|
| 링크 | acmicpc.net/… | 
| 출처 | BOJ | 
| 문제 번호 | 1492 | 
| 문제명 | 합 | 
| 레벨 | 플래티넘 2 | 
| 분류 | 
 수학  | 
	
| 시간복잡도 | O(K^2) | 
| 인풋사이즈 | K<=50 | 
| 사용한 언어 | Python | 
| 제출기록 | 76ms | 
| 최고기록 | 56ms | 
| 해결날짜 | 2021/05/30 | 
풀이
코드
"""Solution code for "BOJ 1492. 합".
- Problem link: https://www.acmicpc.net/problem/1492
- Solution link: http://www.teferi.net/ps/problems/boj/1492
"""
MOD = 1_000_000_007
from teflib import combinatorics
def main():
    N, K = [int(x) for x in input().split()]
    s = [N] + [None] * K
    comb_table = combinatorics.CombTable(K + 1, MOD)
    for i in range(1, K + 1):
        sigma = sum(comb_table.get(i + 1, p) * s[p] for p in range(i))
        s[i] = (pow(N + 1, i + 1, MOD) - 1 - sigma) * pow(i + 1, -1, MOD) % MOD
    print(s[K])
if __name__ == '__main__':
    main()
- Dependency: teflib.combinatorics.CombTable
 
ps/problems/boj/1492.txt · 마지막으로 수정됨: 2023/02/09 16:04 저자 teferi
                
                
토론