結果
問題 | No.389 ロジックパズルの組み合わせ |
ユーザー | 2nanoda series |
提出日時 | 2024-03-17 04:28:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 312 ms / 2,000 ms |
コード長 | 3,081 bytes |
コンパイル時間 | 168 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 228,096 KB |
最終ジャッジ日時 | 2024-09-30 04:38:30 |
合計ジャッジ時間 | 10,913 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 39 ms
52,224 KB |
testcase_01 | AC | 39 ms
51,840 KB |
testcase_02 | AC | 38 ms
51,968 KB |
testcase_03 | AC | 38 ms
52,096 KB |
testcase_04 | AC | 300 ms
228,096 KB |
testcase_05 | AC | 39 ms
51,968 KB |
testcase_06 | AC | 39 ms
52,480 KB |
testcase_07 | AC | 40 ms
52,096 KB |
testcase_08 | AC | 40 ms
51,968 KB |
testcase_09 | AC | 305 ms
167,132 KB |
testcase_10 | AC | 78 ms
88,064 KB |
testcase_11 | AC | 312 ms
173,464 KB |
testcase_12 | AC | 61 ms
72,064 KB |
testcase_13 | AC | 127 ms
107,472 KB |
testcase_14 | AC | 78 ms
87,040 KB |
testcase_15 | AC | 82 ms
88,704 KB |
testcase_16 | AC | 146 ms
135,680 KB |
testcase_17 | AC | 223 ms
153,600 KB |
testcase_18 | AC | 208 ms
146,960 KB |
testcase_19 | AC | 43 ms
52,480 KB |
testcase_20 | AC | 43 ms
52,096 KB |
testcase_21 | AC | 39 ms
51,968 KB |
testcase_22 | AC | 40 ms
52,608 KB |
testcase_23 | AC | 39 ms
52,096 KB |
testcase_24 | AC | 43 ms
57,600 KB |
testcase_25 | AC | 41 ms
52,480 KB |
testcase_26 | AC | 42 ms
52,864 KB |
testcase_27 | AC | 41 ms
52,480 KB |
testcase_28 | AC | 40 ms
52,096 KB |
testcase_29 | AC | 40 ms
52,096 KB |
testcase_30 | AC | 40 ms
52,480 KB |
testcase_31 | AC | 40 ms
52,352 KB |
testcase_32 | AC | 41 ms
52,608 KB |
testcase_33 | AC | 40 ms
52,224 KB |
testcase_34 | AC | 42 ms
52,608 KB |
testcase_35 | AC | 40 ms
52,480 KB |
testcase_36 | AC | 41 ms
52,096 KB |
testcase_37 | AC | 43 ms
52,096 KB |
testcase_38 | AC | 44 ms
52,608 KB |
testcase_39 | AC | 40 ms
52,608 KB |
testcase_40 | AC | 40 ms
52,736 KB |
testcase_41 | AC | 39 ms
52,608 KB |
testcase_42 | AC | 38 ms
52,352 KB |
testcase_43 | AC | 38 ms
52,096 KB |
testcase_44 | AC | 39 ms
52,608 KB |
testcase_45 | AC | 39 ms
52,096 KB |
testcase_46 | AC | 43 ms
58,112 KB |
testcase_47 | AC | 39 ms
52,352 KB |
testcase_48 | AC | 40 ms
52,096 KB |
testcase_49 | AC | 102 ms
119,808 KB |
testcase_50 | AC | 46 ms
61,568 KB |
testcase_51 | AC | 46 ms
61,184 KB |
testcase_52 | AC | 54 ms
68,608 KB |
testcase_53 | AC | 51 ms
63,232 KB |
testcase_54 | AC | 56 ms
70,272 KB |
testcase_55 | AC | 43 ms
58,624 KB |
testcase_56 | AC | 48 ms
62,848 KB |
testcase_57 | AC | 69 ms
84,864 KB |
testcase_58 | AC | 95 ms
111,616 KB |
testcase_59 | AC | 81 ms
96,640 KB |
testcase_60 | AC | 60 ms
74,880 KB |
testcase_61 | AC | 139 ms
128,200 KB |
testcase_62 | AC | 60 ms
76,672 KB |
testcase_63 | AC | 93 ms
112,128 KB |
testcase_64 | AC | 55 ms
70,272 KB |
testcase_65 | AC | 53 ms
66,176 KB |
testcase_66 | AC | 57 ms
74,496 KB |
testcase_67 | AC | 59 ms
72,704 KB |
testcase_68 | AC | 56 ms
71,424 KB |
testcase_69 | AC | 161 ms
147,456 KB |
testcase_70 | AC | 95 ms
98,944 KB |
testcase_71 | AC | 131 ms
122,112 KB |
testcase_72 | AC | 78 ms
84,480 KB |
testcase_73 | AC | 106 ms
108,800 KB |
testcase_74 | AC | 138 ms
129,024 KB |
testcase_75 | AC | 120 ms
115,456 KB |
testcase_76 | AC | 72 ms
81,408 KB |
testcase_77 | AC | 207 ms
171,392 KB |
testcase_78 | AC | 54 ms
65,920 KB |
testcase_79 | AC | 39 ms
51,840 KB |
testcase_80 | AC | 40 ms
51,968 KB |
testcase_81 | AC | 41 ms
51,840 KB |
testcase_82 | AC | 40 ms
51,968 KB |
testcase_83 | AC | 40 ms
51,840 KB |
testcase_84 | AC | 41 ms
51,840 KB |
testcase_85 | AC | 40 ms
52,224 KB |
testcase_86 | AC | 40 ms
52,096 KB |
testcase_87 | AC | 40 ms
52,096 KB |
testcase_88 | AC | 38 ms
52,224 KB |
testcase_89 | AC | 106 ms
109,056 KB |
testcase_90 | AC | 140 ms
114,124 KB |
testcase_91 | AC | 217 ms
146,896 KB |
testcase_92 | AC | 73 ms
80,896 KB |
testcase_93 | AC | 149 ms
119,368 KB |
testcase_94 | AC | 50 ms
62,720 KB |
testcase_95 | AC | 157 ms
119,484 KB |
testcase_96 | AC | 129 ms
105,436 KB |
testcase_97 | AC | 123 ms
108,416 KB |
testcase_98 | AC | 85 ms
91,648 KB |
ソースコード
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)