結果

問題 No.1417 100の倍数かつ正整数(2)
コンテスト
ユーザー nephrologist
提出日時 2021-03-05 21:59:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 205 ms / 3,000 ms
コード長 1,751 bytes
コンパイル時間 161 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 88,248 KB
最終ジャッジ日時 2024-10-07 01:51:08
合計ジャッジ時間 4,696 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

s = input()

keta = len(s)

dp = [[[[0] * (2) for _ in range(3)] for _ in range(3)] for _ in range(keta + 1)]

# dp[keta][*2][*5][smaller]
mod = 10 ** 9 + 7
dp[0][0][0][0] = 1
for i in range(1, keta):
    dp[i][0][0][1] = 1
for i in range(keta):
    ni = i + 1
    num = int(s[i])
    for j in range(3):
        for k in range(3):
            for smaller in range(2):
                if smaller:
                    for d in range(10):
                        if d == 0:
                            continue
                        nj = j
                        nk = k
                        nsmaller = smaller
                        if d == 4 or d == 8:
                            nj = 2
                        elif d % 2 == 0:
                            nj = min(2, j + 1)
                        if d == 5:
                            nk = min(2, k + 1)
                        dp[ni][nj][nk][nsmaller] += dp[i][j][k][smaller]
                        dp[ni][nj][nk][nsmaller] %= mod
                else:
                    for d in range(num + 1):
                        nj = j
                        nk = k
                        nsmaller = 1
                        if d == num:
                            nsmaller = 0
                        if d == 0:
                            continue

                        if d == 4 or d == 8:
                            nj = 2
                        elif d % 2 == 0:
                            nj = min(2, j + 1)
                        if d == 5:
                            nk = min(2, k + 1)
                        dp[ni][nj][nk][nsmaller] += dp[i][j][k][smaller]
                        dp[ni][nj][nk][nsmaller] %= mod

ans = dp[keta][2][2][0] + dp[keta][2][2][1]
print(ans % mod)
0