結果

問題 No.1417 100の倍数かつ正整数(2)
ユーザー H20H20
提出日時 2021-02-20 02:25:30
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 899 bytes
コンパイル時間 423 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 21,792 KB
最終ジャッジ日時 2024-09-25 00:21:36
合計ジャッジ時間 20,106 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import product

N = input()
mod=10**9+7

def count(a):
    n = len(a)
    dp=[[[[[0] * 2 for i in range(3)] for j in range(3)] for k in range(2)] for l in range(n+1)]
    dp[0][0][0][0][0] = 1

    for i, less, divcnt5, divcnt2, has0 in product(range(n), (0,1), range(3), range(3), (0,1)):
        max_d = 9 if less else int(a[i])
        for d in range(max_d+1):
            less_ = less or d < max_d
            divcnt5_ = min(2,divcnt5+(d==5))
            if d==4 or d==8:
                divcnt2_=2
            elif d==2 or d==6:
                divcnt2_=min(2,divcnt2+1)
            else:
                divcnt2_ = divcnt2
            has0_ = has0 or d==0
            dp[i + 1][less_][divcnt5_][divcnt2_][has0_] += dp[i][less][divcnt5][divcnt2][has0]

    return dp[n][0][2][2][0]+dp[n][1][2][2][0]

ans = count(N)
for i in range(len(N)):
    ans += count('9'*i)
print(ans%mod)
0