結果

問題 No.1631 Sorting Integers (Multiple of K) Easy
ユーザー 👑 Kazun
提出日時 2021-08-05 20:34:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 278 ms / 3,000 ms
コード長 538 bytes
コンパイル時間 647 ms
コンパイル使用メモリ 82,392 KB
実行使用メモリ 107,904 KB
最終ジャッジ日時 2024-09-16 15:23:48
合計ジャッジ時間 4,938 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import product
N,K=map(int,input().split())
C=[0]+list(map(int,input().split()))

X=[]
for i in range(10):
    X+=[i]*C[i]

D={}
M=[]
k=0
for t in product(*[range(C[i]+1) for i in range(10)]):
    D[t]=k
    M.append(t)
    k+=1

DP=[[0]*K for _ in range(k)]
DP[0][0]=1

for S in range(k):
    t=M[S]
    l=list(t)
    for r in range(10):
        if t[r]+1<=C[r]:
            l[r]+=1
            T=D[tuple(l)]
            for m in range(K):
                DP[T][(10*m+r)%K]+=DP[S][m]
            l[r]-=1

print(DP[-1][0])
0