結果

問題 No.895 MESE
ユーザー maspy
提出日時 2020-03-18 20:05:34
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 619 ms / 2,000 ms
コード長 1,002 bytes
コンパイル時間 122 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 46,208 KB
最終ジャッジ日時 2024-12-14 02:26:19
合計ジャッジ時間 14,990 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3.8
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
MOD = 10 ** 9 + 7

a, b, c = map(int, read().split())

U = 3 * 10 ** 5 + 10
pow2 = [1] * U
fact = [1] * U
fact_inv = [1] * U
for n in range(1, U):
    pow2[n] = 2 * pow2[n - 1] % MOD
    fact[n] = n * fact[n - 1] % MOD
fact_inv[-1] = pow(fact[-1], MOD - 2, MOD)
for n in range(U - 1, 0, -1):
    fact_inv[n - 1] = n * fact_inv[n] % MOD


def solve(a, b, c):
    a -= 1
    answer = 0
    for n in range(c - 1, a + b + c):
        m = a + b + c - 1 - n
        cnt = fact[n] * fact_inv[c - 1] * fact_inv[n - c + 1] * fact[a + b] * fact_inv[a] * fact_inv[b]
        if a >= m:
            cnt -= fact[n] * fact_inv[b] * fact_inv[a - m] * fact_inv[c - 1]
        if n == 0:
            avg = 1
        else:
            avg = pow2[n] + (pow2[n] - 1) * fact[n - 1] * fact_inv[n] * (c - 1)
        answer += cnt * avg
    return answer % MOD


print(solve(a, b, c))
0