結果
問題 | No.1631 Sorting Integers (Multiple of K) Easy |
ユーザー |
![]() |
提出日時 | 2021-07-07 12:31:18 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,139 ms / 3,000 ms |
コード長 | 1,411 bytes |
コンパイル時間 | 552 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 95,968 KB |
最終ジャッジ日時 | 2024-09-15 21:36:50 |
合計ジャッジ時間 | 13,175 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 28 |
ソースコード
import collections import itertools def f(c,m): C=len(c) c=[int(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=[int(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(str(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=set() for x in all: y=''.join(x) if y in s: continue s.add(y) 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)