結果
問題 | No.1634 Sorting Integers (Multiple of K) Hard |
ユーザー | とりゐ |
提出日時 | 2021-07-30 20:14:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,350 ms / 3,000 ms |
コード長 | 1,433 bytes |
コンパイル時間 | 317 ms |
コンパイル使用メモリ | 87,192 KB |
実行使用メモリ | 183,332 KB |
最終ジャッジ日時 | 2023-10-14 02:41:19 |
合計ジャッジ時間 | 31,065 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 123 ms
149,896 KB |
testcase_01 | AC | 126 ms
149,684 KB |
testcase_02 | AC | 125 ms
149,972 KB |
testcase_03 | AC | 236 ms
156,568 KB |
testcase_04 | AC | 153 ms
156,156 KB |
testcase_05 | AC | 1,122 ms
171,844 KB |
testcase_06 | AC | 1,087 ms
172,096 KB |
testcase_07 | AC | 1,095 ms
170,560 KB |
testcase_08 | AC | 972 ms
173,496 KB |
testcase_09 | AC | 119 ms
149,504 KB |
testcase_10 | AC | 121 ms
150,004 KB |
testcase_11 | AC | 1,276 ms
180,604 KB |
testcase_12 | AC | 1,213 ms
179,664 KB |
testcase_13 | AC | 1,348 ms
182,924 KB |
testcase_14 | AC | 1,234 ms
179,684 KB |
testcase_15 | AC | 1,222 ms
179,384 KB |
testcase_16 | AC | 1,302 ms
183,332 KB |
testcase_17 | AC | 650 ms
164,952 KB |
testcase_18 | AC | 699 ms
166,240 KB |
testcase_19 | AC | 885 ms
170,132 KB |
testcase_20 | AC | 581 ms
165,060 KB |
testcase_21 | AC | 330 ms
158,800 KB |
testcase_22 | AC | 1,173 ms
174,032 KB |
testcase_23 | AC | 1,228 ms
176,520 KB |
testcase_24 | AC | 1,250 ms
178,572 KB |
testcase_25 | AC | 1,254 ms
178,828 KB |
testcase_26 | AC | 1,256 ms
180,280 KB |
testcase_27 | AC | 1,291 ms
180,876 KB |
testcase_28 | AC | 1,296 ms
183,136 KB |
testcase_29 | AC | 1,297 ms
182,360 KB |
testcase_30 | AC | 1,253 ms
182,284 KB |
testcase_31 | AC | 1,350 ms
183,060 KB |
ソースコード
import collections import itertools def f(c,m): C=len(c) c=[i for i in c] dp=[[] for i in range(1<<C)] dp[0].append(0) for i in range(1<<C): j=bin(i).count('1') for k in range(C): if (i>>k)&1==0: for l in dp[i]: dp[i+2**k].append((l+q[m+k][c[j]])%p) cnt=1 for i in collections.Counter(c).values(): cnt*=fact[i] t=collections.Counter(dp[-1]) return t,cnt def f2(c,m): C=len(c) c=[i for i in c] dp=[[] for i in range(1<<C)] dp[0].append(0) for i in range(1<<C): j=bin(i).count('1') for k in range(C): if (i>>k)&1==0: for l in dp[i]: dp[i+2**k].append((l+q[m+k][c[j]])%p) cnt=1 for i in collections.Counter(c).values(): cnt*=fact[i] return dp[-1],cnt n,p=map(int,input().split()) c=list(map(int,input().split())) a=[] for i in range(9): for j in range(c[i]): a.append(i+1) all=tuple(itertools.combinations(a,n//2)) ans=0 q=[[0]*10 for i in range(n+1)] for i in range(n+1): for j in range(10): q[i][j]=pow(10,i,p)*j%p fact=[1]*(n+1) for i in range(1,n+1): fact[i]=fact[i-1]*i s=[0]*10**7 for x in all: y=0 for i in range(n//2): y+=x[i]*pow(10,i) if s[y]: continue s[y]=1 b=a.copy() for i in x: b.remove(i) d1,n1=f2(x,(n+1)//2) d2,n2=f(b,0) n3=n1*n2 num=0 for i in d1: if i==0: if 0 in d2: num+=d2[0] elif p-i in d2: num+=d2[p-i] ans+=num//n3 print(ans)