結果

問題 No.1631 Sorting Integers (Multiple of K) Easy
ユーザー lloyz
提出日時 2021-12-27 23:28:30
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 870 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 82,112 KB
実行使用メモリ 207,004 KB
最終ジャッジ日時 2024-10-01 10:13:27
合計ジャッジ時間 5,692 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 10 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

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] * 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):
            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