結果

問題 No.1181 Product Sum for All Subsets
コンテスト
ユーザー ntuda
提出日時 2025-03-08 17:16:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 75 ms / 2,000 ms
コード長 520 bytes
コンパイル時間 386 ms
コンパイル使用メモリ 82,388 KB
実行使用メモリ 78,588 KB
最終ジャッジ日時 2025-03-08 17:16:15
合計ジャッジ時間 3,498 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 10 ** 9 + 7
N,K = map(int,input().split())
fact = [1]*(N+1)
rfact = [1]*(N+1)
r = 1
for i in range(1, N+1):
    fact[i] = r = r * i % MOD
rfact[N] = r = pow(fact[N], -1, MOD)
for i in range(N, 0, -1):
    rfact[i-1] = r = r * i % MOD

# nCk (mod MOD) を求める
def comb(n, k):
    return fact[n] * rfact[k] * rfact[n-k] % MOD

KN = pow(K,N,MOD)
avg = ((K * (K + 1) // 2) * pow(K,-1,MOD)) % MOD
ans = 0
TKN = KN
for i in range(N):
    ans += TKN * comb(N,i)
    ans %= MOD
    TKN = (TKN * avg) % MOD
print(ans)
0