結果
問題 |
No.1631 Sorting Integers (Multiple of K) Easy
|
ユーザー |
👑 |
提出日時 | 2022-04-15 17:19:19 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,455 ms / 3,000 ms |
コード長 | 847 bytes |
コンパイル時間 | 318 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 205,056 KB |
最終ジャッジ日時 | 2024-12-24 21:29:38 |
合計ジャッジ時間 | 22,783 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 28 |
ソースコード
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 = [[0] * K 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 in range(K): v = dp[bit ^ (1 << i)][k] dp[bit][(x + k) % K] += v ans = dp[-1][0] fact = [1] for i in range(1, n + 1): fact.append(fact[-1] * i) for c in C: ans //= fact[c] print(ans)