結果

問題 No.1417 100の倍数かつ正整数(2)
ユーザー ygd.
提出日時 2021-03-05 22:00:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 369 ms / 3,000 ms
コード長 1,515 bytes
コンパイル時間 265 ms
コンパイル使用メモリ 82,444 KB
実行使用メモリ 90,668 KB
最終ジャッジ日時 2024-10-07 01:53:13
合計ジャッジ時間 6,446 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

def TWO(X):
    if X == 8:
        return 3
    elif X == 4:
        return 2
    elif X == 2 or X == 6:
        return 1
    else:
        return 0

def FIVE(X):
    if X == 5:
        return 1
    else:
        return 0

NS = str(input()); MOD = pow(10,9)+7
N = len(NS) #桁数
#dp[i][p][q][j]: i桁目まで見て、2のp乗で割れる、5のq乗で割れる、j=0一致/1以下しているかどうかで
dp = [[[[0]*2 for _ in range(3)] for _ in range(3)] for _ in range(N+1)]
dp[0][0][0][0] = 1
for i in range(N):
    for num in range(1,10): #0が一つでもあったらダメなので不要
        for p in range(3):
            for q in range(3):
                np = min(p + TWO(num),2)
                nq = min(q + FIVE(num),2)
                
                if num == int(NS[i]): #一致している場合 0-> 0
                    #print("A",i+1,np,nq,p,q)
                    dp[i+1][np][nq][0] += dp[i][p][q][0]
                    dp[i+1][np][nq][0] %= MOD
                elif num < int(NS[i]): #0 -> 1
                    #print("B",i+1,np,nq,p,q)
                    dp[i+1][np][nq][1] += dp[i][p][q][0]
                    dp[i+1][np][nq][1] %= MOD
                
                dp[i+1][np][nq][1] += dp[i][p][q][1] #1 -> 1
                dp[i+1][np][nq][1] %= MOD
        if i != 0: #最上位以外で
            np = min(TWO(num),2)
            nq = min(FIVE(num),2)
            dp[i+1][np][nq][1] += 1
#print(dp)
ans = dp[N][2][2][0] + dp[N][2][2][1]
print(ans%MOD)
0