結果

問題 No.372 It's automatic
ユーザー takeya_okino
提出日時 2019-06-22 18:09:33
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 1,130 bytes
コンパイル時間 5,136 ms
コンパイル使用メモリ 77,476 KB
実行使用メモリ 1,470,256 KB
最終ジャッジ日時 2024-12-26 09:40:45
合計ジャッジ時間 101,516 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample MLE * 3
other WA * 10 TLE * 3 MLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.util.*;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    String s = sc.next();
    int M = sc.nextInt();
    long MOD = (long)Math.pow(10, 9) + 7;
    int len = s.length();
    long[][][] dp = new long[len][M][2];
    int first = Integer.parseInt(String.valueOf(s.charAt(0)));
    if(first == 0) {
      dp[0][0][0] = 1;
    } else {
      dp[0][first % M][1] = 1;
      dp[0][0][0] = 1;
    }

    for(int i = 1; i < len; i++) {
      int d = Integer.parseInt(String.valueOf(s.charAt(i)));
      for(int j = 0; j < M; j++) {
        if(d > 0) dp[i][d % M][1] = (dp[i][d % M][1] + dp[i - 1][j][0]) % MOD;
        dp[i][j][0] = (dp[i][j][0] + dp[i - 1][j][0]) % MOD;
        dp[i][(10 * j + d) % M][1] = (dp[i][(10 * j + d) % M][1] + dp[i - 1][j][1]) % MOD;
        dp[i][(10 * j) % M][1] = (dp[i][(10 * j) % M][1] + dp[i - 1][j][1]) % MOD;
      }
    }

    long ans = 0;
    long kosuu = 0;
    for(int i = 0; i < s.length(); i++) {
      if(s.charAt(i) == '0') kosuu++;
    }
    ans = (dp[len - 1][0][1] + kosuu) % MOD;
    System.out.println(ans);
  }
}
0