結果

問題 No.1417 100の倍数かつ正整数(2)
ユーザー marroncastle
提出日時 2021-03-06 00:04:30
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,329 bytes
コンパイル時間 310 ms
コンパイル使用メモリ 82,292 KB
実行使用メモリ 122,968 KB
最終ジャッジ日時 2024-10-07 06:23:25
合計ジャッジ時間 11,557 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 N+[0]] for _ in range(3)] for _ in range(3)]
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[j][k][i][l] %= MOD
        dp[j][k][i+1][l] += dp[j][k][i+1][l-1]
        dp[j][k][i+1][l] += dp[j][k][i][-1]*(int(l%2==1)-int(l%5==0))
        dp[min(2,j+1)][k][i+1][l] += dp[j][k][i][-1]*(int(l%2==0)-int(l%4==0))
        dp[min(2,j+2)][k][i+1][l] += dp[j][k][i][-1]*int(l%4==0)
        dp[j][min(2,k+1)][i+1][l] += dp[j][k][i][-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[-1][-1][i+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[j][k][i][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[j][k][i][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[j][k][i][N[i-1]-1]+dp1[i][j][k])*int(N[i]%4==0)
        dp1[i+1][j][min(2,k+1)] += (dp[j][k][i][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