結果

問題 No.389 ロジックパズルの組み合わせ
ユーザー 2nanoda series2nanoda series
提出日時 2024-03-17 04:28:39
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 325 ms / 2,000 ms
コード長 3,081 bytes
コンパイル時間 382 ms
コンパイル使用メモリ 81,700 KB
実行使用メモリ 227,952 KB
最終ジャッジ日時 2024-03-17 04:28:51
合計ジャッジ時間 11,960 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
53,460 KB
testcase_01 AC 39 ms
53,460 KB
testcase_02 AC 39 ms
53,460 KB
testcase_03 AC 39 ms
53,460 KB
testcase_04 AC 321 ms
227,952 KB
testcase_05 AC 39 ms
53,460 KB
testcase_06 AC 60 ms
53,460 KB
testcase_07 AC 38 ms
53,460 KB
testcase_08 AC 38 ms
53,460 KB
testcase_09 AC 317 ms
167,092 KB
testcase_10 AC 76 ms
88,492 KB
testcase_11 AC 325 ms
173,512 KB
testcase_12 AC 56 ms
73,592 KB
testcase_13 AC 129 ms
107,000 KB
testcase_14 AC 102 ms
88,668 KB
testcase_15 AC 79 ms
89,596 KB
testcase_16 AC 159 ms
135,956 KB
testcase_17 AC 245 ms
154,004 KB
testcase_18 AC 211 ms
146,556 KB
testcase_19 AC 39 ms
53,460 KB
testcase_20 AC 38 ms
53,460 KB
testcase_21 AC 39 ms
53,460 KB
testcase_22 AC 40 ms
53,460 KB
testcase_23 AC 38 ms
53,460 KB
testcase_24 AC 41 ms
59,216 KB
testcase_25 AC 44 ms
53,460 KB
testcase_26 AC 40 ms
53,460 KB
testcase_27 AC 38 ms
53,460 KB
testcase_28 AC 40 ms
53,460 KB
testcase_29 AC 38 ms
53,460 KB
testcase_30 AC 39 ms
53,460 KB
testcase_31 AC 39 ms
53,460 KB
testcase_32 AC 38 ms
53,460 KB
testcase_33 AC 39 ms
53,460 KB
testcase_34 AC 40 ms
53,460 KB
testcase_35 AC 41 ms
53,460 KB
testcase_36 AC 39 ms
53,460 KB
testcase_37 AC 38 ms
53,460 KB
testcase_38 AC 39 ms
53,460 KB
testcase_39 AC 39 ms
53,460 KB
testcase_40 AC 39 ms
53,460 KB
testcase_41 AC 39 ms
53,460 KB
testcase_42 AC 39 ms
53,460 KB
testcase_43 AC 39 ms
53,460 KB
testcase_44 AC 39 ms
53,460 KB
testcase_45 AC 38 ms
53,460 KB
testcase_46 AC 67 ms
59,216 KB
testcase_47 AC 39 ms
53,460 KB
testcase_48 AC 39 ms
53,460 KB
testcase_49 AC 95 ms
119,684 KB
testcase_50 AC 44 ms
61,904 KB
testcase_51 AC 44 ms
61,920 KB
testcase_52 AC 51 ms
69,772 KB
testcase_53 AC 46 ms
63,404 KB
testcase_54 AC 53 ms
70,928 KB
testcase_55 AC 43 ms
59,588 KB
testcase_56 AC 45 ms
62,928 KB
testcase_57 AC 65 ms
86,808 KB
testcase_58 AC 93 ms
112,700 KB
testcase_59 AC 78 ms
98,028 KB
testcase_60 AC 57 ms
76,744 KB
testcase_61 AC 159 ms
128,192 KB
testcase_62 AC 58 ms
78,340 KB
testcase_63 AC 90 ms
112,888 KB
testcase_64 AC 51 ms
70,804 KB
testcase_65 AC 48 ms
67,864 KB
testcase_66 AC 54 ms
76,752 KB
testcase_67 AC 53 ms
73,224 KB
testcase_68 AC 53 ms
72,008 KB
testcase_69 AC 179 ms
148,080 KB
testcase_70 AC 100 ms
100,512 KB
testcase_71 AC 142 ms
122,016 KB
testcase_72 AC 83 ms
85,044 KB
testcase_73 AC 119 ms
109,728 KB
testcase_74 AC 146 ms
130,928 KB
testcase_75 AC 131 ms
115,744 KB
testcase_76 AC 73 ms
82,356 KB
testcase_77 AC 222 ms
172,092 KB
testcase_78 AC 51 ms
67,508 KB
testcase_79 AC 39 ms
53,460 KB
testcase_80 AC 38 ms
53,460 KB
testcase_81 AC 39 ms
53,460 KB
testcase_82 AC 38 ms
53,460 KB
testcase_83 AC 39 ms
53,460 KB
testcase_84 AC 39 ms
53,460 KB
testcase_85 AC 38 ms
53,460 KB
testcase_86 AC 40 ms
53,460 KB
testcase_87 AC 39 ms
53,460 KB
testcase_88 AC 38 ms
53,460 KB
testcase_89 AC 99 ms
109,148 KB
testcase_90 AC 139 ms
114,144 KB
testcase_91 AC 216 ms
146,764 KB
testcase_92 AC 66 ms
81,524 KB
testcase_93 AC 148 ms
119,284 KB
testcase_94 AC 49 ms
64,188 KB
testcase_95 AC 160 ms
119,268 KB
testcase_96 AC 126 ms
105,608 KB
testcase_97 AC 119 ms
108,308 KB
testcase_98 AC 79 ms
92,864 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

m=int(input())
h=list(map(int,input().split()))


class Mod:
    # maxdiv ... 割る数の最大値
    # もし、m回maxdiv以下の値で割る場合、
    # 事前計算によりコストmaxdiv+mで計算可能
    # もし、log(mod)*mより早くなるなら使うべき。
    # 未実装。
    def __init__(self,mod:int,maxdiv=None):
        self.mod=mod
        self.facs=[1,1]
        self.invfacs=[1,1]
        self.invs=[None,1]
        if maxdiv!=None:
            raise NotImplementedError()
            # self.maxdiv=maxdiv
    def __call__(self,a):
        return a%self.mod
    def add(self,a,b):
        return self(a+b)
    def sub(self,a,b):
        return self(a-b)
    def mul(self,a,b):
        return self(a*b)
    # a^n mod mod # 使用条件　なし O(log n)
    def pow(self,a,n):
        return pow(a,n,self.mod)
    # a/b mod mod (使用条件 : modは素数,aとmodは互いに素) , O(log p)
    def truediv(self,a,b):
        return self(a*self.inv(b))
    # a^{-1} mod mod (使用条件 : modは素数,aとmodは互いに素) , O(log p)
    def inv(self,a):
        return self.pow(a,self.mod-2) # フェルマーの小定理

    # [0! 1! 2! ... n!] , [(0!)^{-1},...,(n!)^{-1}] , [None,1^-1,...,n^-1] mod mod を計算． 
    # 使用条件 mod が 1,...,n と互いに疎
    # O(n)
    # 疑問点 条件は正しいか，Noneは正しいか
    def calc_fac_invfac(self,max_n:int):
        mod=self.mod
        facs=self.facs
        invs=self.invs
        invfacs=self.invfacs
        n=max_n
        start=len(self.facs)
        for i in range(start,n+1):
            facs.append(self(facs[i - 1] * i))
            invs.append( mod - self(invs[mod%i] * (mod // i)))
            invfacs.append(self(invfacs[i - 1] * invs[i]))
        

    # calc_fac_invfac を先に使用すること．
    # 事前計算をすれば計算コストはO(1)
    # していないならばO(n)
    def nCk(self,n:int,k:int,cached=True):
        # cached = Trueの時、計算コストは最低O(n)
        if (n < k):
            return 0
        if (n < 0 or k < 0):
            return 0
        if cached:
            if len(self.facs)<n+1:
                self.calc_fac_invfac(max_n=n)
            return self(self.facs[n] * (self(self.invfacs[k] * self.invfacs[n - k])))
        else: # cached = Falseの時、計算コストは O(k)
            r=1
            for ki in range(k):
                r=mod.mul(r,n-ki)
                r=mod.truediv(r,k-ki)
            return r
mod=Mod(10**9+7)

# ますは10^6以下である。
total=sum(h)
if total==0:
    print(1)
    exit(0)
if total+len(h)-1>m:
    print("NA")
    exit(0)

n=len(h)
m=m-total+n
# h=[1]*len(h)
# mますで,1のみで構成される

r=0

m1=m-n
n1=n
# 右端を使わない場合
# 各々のブロックは必ず右が空洞であるとして良い。
# したがって、
# m1の中からnを選ぶ
r=mod.add(r,mod.nCk(m1,n1))
# print(m1,n1,r)

# 右端を使う場合
# 右端を除くとさっきと同じ。
m2=m-n
n2=n-1
r=mod.add(r,mod.nCk(m2,n2))
# print(m2,n2,r)

print(r)
0