結果

問題 No.315 世界のなんとか3.5
ユーザー takeya_okinotakeya_okino
提出日時 2019-07-01 19:44:56
言語 Java21
(openjdk 21)
結果
TLE  
実行時間 -
コード長 3,472 bytes
コンパイル時間 2,192 ms
コンパイル使用メモリ 75,032 KB
実行使用メモリ 262,172 KB
最終ジャッジ日時 2023-09-25 03:37:36
合計ジャッジ時間 8,567 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 126 ms
56,076 KB
testcase_01 AC 130 ms
55,652 KB
testcase_02 AC 212 ms
60,760 KB
testcase_03 AC 126 ms
55,620 KB
testcase_04 AC 128 ms
55,972 KB
testcase_05 AC 134 ms
55,516 KB
testcase_06 AC 210 ms
61,304 KB
testcase_07 AC 131 ms
55,660 KB
testcase_08 AC 142 ms
56,052 KB
testcase_09 AC 128 ms
55,376 KB
testcase_10 AC 128 ms
55,720 KB
testcase_11 AC 126 ms
55,840 KB
testcase_12 TLE -
testcase_13 MLE -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

public class Main {
  public static long MOD = (long)Math.pow(10, 9) + 7;
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    String A = sc.next();
    String B = sc.next();
    int p = sc.nextInt();
    long ans = (func(B, p) - func(A, p) + MOD) % MOD;
    int m1 = 0;
    int kosuu = 0;
    int m2 = 0;
    for(int i = 0; i < A.length(); i++) {
      int d = Integer.parseInt(String.valueOf(A.charAt(i)));
      if(d == 3) kosuu++;
      m1 = (m1 + d) % 3;
      m2 = (10 * m2 + d) % p;
    }
    if(((m1 == 0) || (kosuu > 0)) && (m2 != 0)) ans = (ans + 1) % MOD;
    System.out.println(ans);
  }

  public static long func(String X, int p) {
    int len = X.length();
    long[][][][][] dp = new long[len][3][2][p][2];
    int first = Integer.parseInt(String.valueOf(X.charAt(0)));
    dp[0][0][0][0][1]++;
    if(first == 3) {
      dp[0][(first) % 3][1][(first) % p][0]++;
    } else {
      dp[0][(first) % 3][0][(first) % p][0]++;
    }
    for(int i = 1; i < first; i++) {
      if(i == 3) {
        dp[0][i % 3][1][i % p][1]++;
      } else {
        dp[0][i % 3][0][i % p][1]++;
      }      
    }

    for(int i = 1; i < len; i++) {
      int d = Integer.parseInt(String.valueOf(X.charAt(i)));
      for(int j = 0; j < 3; j++) {
        for(int k = 0; k < p; k++) {
          for(int l = 0; l <= d; l++) {
            if(l == d) {
              if(l == 3) {
                dp[i][(j + l) % 3][1][(10 * k + l) % p][0] = (dp[i][(j + l) % 3][1][(10 * k + l) % p][0] + dp[i - 1][j][0][k][0]) % MOD;
              } else {
                dp[i][(j + l) % 3][0][(10 * k + l) % p][0] = (dp[i][(j + l) % 3][0][(10 * k + l) % p][0] + dp[i - 1][j][0][k][0]) % MOD;
              }
            } else {
              if(l == 3) {
                dp[i][(j + l) % 3][1][(10 * k + l) % p][1] = (dp[i][(j + l) % 3][1][(10 * k + l) % p][1] + dp[i - 1][j][0][k][0]) % MOD;
              } else {
                dp[i][(j + l) % 3][0][(10 * k + l) % p][1] = (dp[i][(j + l) % 3][0][(10 * k + l) % p][1] + dp[i - 1][j][0][k][0]) % MOD;
              }              
            }
          }
          for(int l = 0; l <= 9; l++) {
            if(l == 3) {
              dp[i][(j + l) % 3][1][(10 * k + l) % p][1] = (dp[i][(j + l) % 3][1][(10 * k + l) % p][1] + dp[i - 1][j][0][k][1]) % MOD;
            } else {
              dp[i][(j + l) % 3][0][(10 * k + l) % p][1] = (dp[i][(j + l) % 3][0][(10 * k + l) % p][1] + dp[i - 1][j][0][k][1]) % MOD;
            }
          }
          for(int l = 0; l <= d; l++) {
            if(l == d) {
              dp[i][(j + l) % 3][1][(10 * k + l) % p][0] = (dp[i][(j + l) % 3][1][(10 * k + l) % p][0] + dp[i - 1][j][1][k][0]) % MOD;
            } else {
              dp[i][(j + l) % 3][1][(10 * k + l) % p][1] = (dp[i][(j + l) % 3][1][(10 * k + l) % p][1] + dp[i - 1][j][1][k][0]) % MOD;
            }
          }
          for(int l = 0; l <= 9; l++) {
            dp[i][(j + l) % 3][1][(10 * k + l) % p][1] = (dp[i][(j + l) % 3][1][(10 * k + l) % p][1] + dp[i - 1][j][1][k][1]) % MOD;
          }
        }
      }
    }
    long ret = 0;
    for(int k = 1; k < p; k++) {
      for(int l = 0; l < 2; l++) {
        ret = (ret + dp[len - 1][0][0][k][l]) % MOD;
      }
    }
    for(int k = 1; k < p; k++) {
      for(int l = 0; l < 2; l++) {
        for(int j = 0; j < 3; j++) {
          ret = (ret + dp[len - 1][j][1][k][l]) % MOD;
        }
      }
    }
    return ret;
  }
}
0