結果

問題 No.389 ロジックパズルの組み合わせ
ユーザー titia
提出日時 2023-02-08 01:54:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 349 ms / 2,000 ms
コード長 508 bytes
コンパイル時間 168 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 256,016 KB
最終ジャッジ日時 2024-07-05 18:24:51
合計ジャッジ時間 30,555 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 99
権限があれば一括ダウンロードができます

ソースコード

diff #

M=int(input())
H=list(map(int,input().split()))

if H==[0]:
    print(1)
    exit()

mod=10**9+7

FACT=[1]
for i in range(1,2*10**6+1):
    FACT.append(FACT[-1]*i%mod)

FACT_INV=[pow(FACT[-1],mod-2,mod)]
for i in range(2*10**6,0,-1):
    FACT_INV.append(FACT_INV[-1]*i%mod)

FACT_INV.reverse()

def Combi(a,b):
    if 0<=b<=a:
        return FACT[a]*FACT_INV[b]%mod*FACT_INV[a-b]%mod
    else:
        return 0

x=sum(H)+len(H)-1
rest=M-x

if rest<0:
    print("NA")
else:
    print(Combi(rest+len(H),rest))
0