結果

問題 No.389 ロジックパズルの組み合わせ
コンテスト
ユーザー Chihaya_chan
提出日時 2020-08-10 03:21:04
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 438 ms / 2,000 ms
コード長 922 bytes
コンパイル時間 303 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 66,124 KB
最終ジャッジ日時 2024-10-06 00:55:10
合計ジャッジ時間 11,128 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 99
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Combi():
    def __init__(self, N, mod):
        self.power = [1 for _ in range(N+1)]
        self.rev = [1 for _ in range(N+1)]
        self.mod = mod
        for i in range(2, N+1):
            self.power[i] = (self.power[i-1]*i) % self.mod
        self.rev[N] = pow(self.power[N], self.mod-2, self.mod)
        for j in range(N, 0, -1):
            self.rev[j-1] = (self.rev[j]*j) % self.mod

    def C(self, K, R):
        if K < R:
            return "NA"
        else:
            return ((self.power[K])*(self.rev[K-R])*(self.rev[R])) % self.mod

    def P(self, K, R):
        if K < R:
            return "NA"
        else:
            return (self.power[K])*(self.rev[K-R]) % self.mod

mod =10**9 + 7
M = int(input())
H = list(map(int, input().split()))
L = len(H)
M += (L-sum(H))
M -= (L-1)

if sum(H)==0:
    print(1)
    exit()
if M >= 0:
    c = Combi(M, mod)
    print(c.C(M, L))
else:
    print("NA")
0