結果

問題 No.1417 100の倍数かつ正整数(2)
ユーザー marroncastle
提出日時 2021-03-06 00:01:05
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,328 bytes
コンパイル時間 231 ms
コンパイル使用メモリ 82,004 KB
実行使用メモリ 125,180 KB
最終ジャッジ日時 2024-10-07 06:16:40
合計ジャッジ時間 11,126 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31 TLE * 1 -- * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 10**9+7

N = list(map(int,list(input())))[::-1]
digit = len(N)
dp = [[[[0]*10 for _ in range(3)] for _ in range(3)] for _ in N+[0]]
dp[0][0][0] = [1]*10
for i in range(digit): #配るDP
  for j in range(3): #2
    for k in range(3): #5
      for l in range(1,10):
        dp[i][j][k][l] %= MOD
        dp[i+1][j][k][l] += dp[i+1][j][k][l-1]
        dp[i+1][j][k][l] += dp[i][j][k][-1]*(int(l%2==1)-int(l%5==0))
        dp[i+1][min(2,j+1)][k][l] += dp[i][j][k][-1]*(int(l%2==0)-int(l%4==0))
        dp[i+1][min(2,j+2)][k][l] += dp[i][j][k][-1]*int(l%4==0)
        dp[i+1][j][min(2,k+1)][l] += dp[i][j][k][-1]*int(l%5==0)
dp1 = [[[0]*3 for _ in range(3)] for _ in N+[0]]
ans = 0
for i in range(digit-1):
  ans += dp[i+1][-1][-1][-1]
  ans %= MOD
for i in range(digit):
  if N[i-1]>0 and N[i]>0:
    for j in range(3): #2
      for k in range(3): #5
        dp1[i][j][k] %= MOD
        dp1[i+1][j][k] += (dp[i][j][k][N[i-1]-1]+dp1[i][j][k])*(int(N[i]%2==1)-int(N[i]%5==0))
        dp1[i+1][min(2,j+1)][k] += (dp[i][j][k][N[i-1]-1]+dp1[i][j][k])*(int(N[i]%2==0)-int(N[i]%4==0))
        dp1[i+1][min(2,j+2)][k] += (dp[i][j][k][N[i-1]-1]+dp1[i][j][k])*int(N[i]%4==0)
        dp1[i+1][j][min(2,k+1)] += (dp[i][j][k][N[i-1]-1]+dp1[i][j][k])*int(N[i]%5==0)
ans += dp1[-1][-1][-1]+dp[-1][-1][-1][N[-1]-1]
print(ans%MOD)
#   dp1[i+1]
0