結果
| 問題 |
No.389 ロジックパズルの組み合わせ
|
| コンテスト | |
| ユーザー |
shinchan
|
| 提出日時 | 2020-10-26 23:54:08 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 173 ms / 2,000 ms |
| コード長 | 540 bytes |
| コンパイル時間 | 353 ms |
| コンパイル使用メモリ | 82,484 KB |
| 実行使用メモリ | 149,440 KB |
| 最終ジャッジ日時 | 2024-07-21 21:45:42 |
| 合計ジャッジ時間 | 11,935 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| other | AC * 99 |
ソースコード
m = int(input())
a = list(map(int,input().split()))
N = 1000010
mod = 1000000007
fac = [1] * N
finv = [1] * N
inv = [1] * N
for i in range(2, N):
fac[i] = fac[i - 1] * i % mod
inv[i] = mod - inv[mod % i] * (mod // i) % mod
finv[i] = finv[i - 1] * inv[i] % mod
def com(n, k):
if n < k:
return 0
if n < 0 or k < 0:
return 0
return fac[n]*(finv[k]*finv[n - k]%mod) %mod
def calc(n, k):
return com(n + k - 1, n)
s = 0
if a[0] != 0:
s = len(a)
ans = calc(m - (sum(a) + s - 1), s + 1)
if ans == 0:
print("NA")
else:
print(ans)
shinchan