結果

問題 No.1631 Sorting Integers (Multiple of K) Easy
ユーザー lloyzlloyz
提出日時 2021-12-27 23:26:27
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 940 bytes
コンパイル時間 218 ms
コンパイル使用メモリ 81,796 KB
実行使用メモリ 200,488 KB
最終ジャッジ日時 2024-10-01 10:11:14
合計ジャッジ時間 8,435 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 11 TLE * 1 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    n, k = map(int, input().split())
    C = list(map(int, input().split()))
    L = []
    for i in range(9):
        for num in range(C[i]):
            L.append(i + 1)

    popcount = [0 for _ in range(200000)]
    for i in range(1, 200000):
        popcount[i] = popcount[i // 2] + i % 2

    fact = [1]
    for i in range(1, n + 1):
        fact.append(fact[-1] * i)

    DP = [[0] * k for _ in range(1 << n)]
    DP[0][0] = 1
    for bit in range(1 << n):
        digit = popcount[bit]
        for i in range(k):
            if DP[bit][i] != 0:
                for j in range(n):
                    if (bit >> j) & 1:
                        continue
                    num = pow(10, digit) * L[j] % k
                    r = (i + num) % k
                    DP[bit | (1 << j)][r] += DP[bit][i]
    ans = DP[-1][0]
    for i in range(9):
        ans //= fact[C[i]]
    print(ans)

if __name__ == '__main__':
    main()
0