結果

問題 No.372 It's automatic
ユーザー lam6er
提出日時 2025-03-31 17:25:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 4,614 ms / 6,000 ms
コード長 1,049 bytes
コンパイル時間 377 ms
コンパイル使用メモリ 82,360 KB
実行使用メモリ 260,464 KB
最終ジャッジ日時 2025-03-31 17:27:28
合計ジャッジ時間 58,081 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 10**9 + 7

S = input().strip()
M = int(input())
mod = M

prev_dp0 = 1
prev_dp1 = [0] * mod
prev_dp2 = [0] * mod

for c in S:
    c_num = int(c)
    current_mod = c_num % mod
    
    new_dp0 = prev_dp0
    new_dp1 = prev_dp1.copy()
    new_dp2 = prev_dp2.copy()
    
    # Transition from prev_dp0
    if c_num == 0:
        new_dp2[current_mod] = (new_dp2[current_mod] + prev_dp0) % MOD
    else:
        new_dp1[current_mod] = (new_dp1[current_mod] + prev_dp0) % MOD
    
    # Transition from prev_dp1 (appending current character)
    temp_dp1 = [0] * mod
    for r in range(mod):
        if prev_dp1[r]:
            new_r = (r * 10 + c_num) % mod
            temp_dp1[new_r] = (temp_dp1[new_r] + prev_dp1[r]) % MOD
    
    for r in range(mod):
        new_dp1[r] = (new_dp1[r] + temp_dp1[r]) % MOD
    
    # Update previous states for next iteration
    prev_dp0, prev_dp1, prev_dp2 = new_dp0, new_dp1, new_dp2

# Sum the valid subsequences: those in dp1 and dp2 with remainder 0
answer = (prev_dp1[0] + prev_dp2[0]) % MOD
print(answer)
0