結果
問題 |
No.1631 Sorting Integers (Multiple of K) Easy
|
ユーザー |
|
提出日時 | 2022-04-15 17:17:45 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 854 bytes |
コンパイル時間 | 229 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 840,192 KB |
最終ジャッジ日時 | 2024-12-24 21:28:22 |
合計ジャッジ時間 | 43,541 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 RE * 1 MLE * 1 |
other | AC * 12 RE * 8 TLE * 6 MLE * 2 |
ソースコード
from math import gcd def popcount(x): x = x - ((x >> 1) & 0x55555555) x = (x & 0x33333333) + ((x >> 2) & 0x33333333) x = (x + (x >> 4)) & 0x0f0f0f0f x += x >> 8 x += x >> 16 return x & 0x0000003f n, K = map(int, input().split()) C = list(map(int, input().split())) lst = [] for i, c in enumerate(C, 1): lst += [i] * c ten = [1] for _ in range(n): ten.append(ten[-1] * 10 % K) dp = [{} for _ in range(1 << n)] dp[0][0] = 1 for bit in range(1, 1 << n): pc = popcount(bit) for i in range(n): if bit >> i & 1: x = lst[i] * ten[pc - 1] % K for k, v in dp[bit ^ (1 << i)].items(): dp[bit][(x + k) % K] = dp[bit].get((x + k) % K, 0) + v ans = dp[-1].get(0) fact = [1] for i in range(1, n + 1): fact.append(fact[-1] * i) for c in C: ans //= fact[c] print(ans)