結果

問題 No.1631 Sorting Integers (Multiple of K) Easy
ユーザー gew1fw
提出日時 2025-06-12 16:04:09
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,109 bytes
コンパイル時間 186 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 319,736 KB
最終ジャッジ日時 2025-06-12 16:04:42
合計ジャッジ時間 5,967 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 10 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict

def main():
    import sys
    input = sys.stdin.read().split()
    idx = 0
    N = int(input[idx])
    idx += 1
    K = int(input[idx])
    idx += 1
    c = list(map(int, input[idx:idx+9]))
    idx += 9

    digits = [i for i in range(1, 10) if c[i-1] > 0]
    if not digits:
        print(0)
        return
    md = len(digits)
    max_counts = [c[i-1] for i in digits]

    dp = defaultdict(int)
    initial_counts = tuple([0] * md)
    dp[(0, initial_counts)] = 1

    for _ in range(N):
        new_dp = defaultdict(int)
        for (rem, counts), cnt in dp.items():
            for i in range(md):
                if counts[i] < max_counts[i]:
                    new_counts = list(counts)
                    new_counts[i] += 1
                    new_counts = tuple(new_counts)
                    new_rem = (rem * 10 + digits[i]) % K
                    new_dp[(new_rem, new_counts)] += cnt
        dp = new_dp

    ans = 0
    for (rem, counts), cnt in dp.items():
        if rem == 0:
            ans += cnt
    print(ans)

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