結果

問題 No.315 世界のなんとか3.5
ユーザー takeya_okinotakeya_okino
提出日時 2019-07-01 22:04:47
言語 Java21
(openjdk 21)
結果
TLE  
実行時間 -
コード長 5,426 bytes
コンパイル時間 1,944 ms
コンパイル使用メモリ 79,216 KB
実行使用メモリ 461,076 KB
最終ジャッジ日時 2024-07-18 07:26:26
合計ジャッジ時間 16,398 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 114 ms
41,244 KB
testcase_01 AC 125 ms
41,492 KB
testcase_02 AC 191 ms
48,584 KB
testcase_03 AC 100 ms
41,380 KB
testcase_04 AC 103 ms
41,504 KB
testcase_05 AC 111 ms
40,532 KB
testcase_06 AC 184 ms
48,696 KB
testcase_07 AC 124 ms
41,388 KB
testcase_08 AC 123 ms
42,500 KB
testcase_09 AC 109 ms
40,744 KB
testcase_10 AC 118 ms
41,544 KB
testcase_11 AC 120 ms
41,332 KB
testcase_12 AC 1,262 ms
243,924 KB
testcase_13 AC 1,253 ms
243,568 KB
testcase_14 TLE -
testcase_15 TLE -
testcase_16 TLE -
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) {
      if((len - 5) <= 0) {
        dp[0][(first) % 3][1][(first) % p][0]++;
      } else {
        dp[0][(first) % 3][1][0][0]++;
      }
    } else {
      if((len - 5) <= 0) {
        dp[0][(first) % 3][0][(first) % p][0]++;
      } else {
        dp[0][(first) % 3][0][0][0]++;
      }
    }
    for(int i = 1; i < first; i++) {
      if(i == 3) {
        if((len - 5) <= 0) {
          dp[0][i % 3][1][i % p][1]++;
        } else {
          dp[0][i % 3][1][0][1]++;
        }
      } else {
        if((len - 5) <= 0) {
          dp[0][i % 3][0][i % p][1]++;
        } else {
          dp[0][i % 3][0][0][1]++;
        }
      }      
    }

    for(int i = 1; i < len; i++) {
      int d = Integer.parseInt(String.valueOf(X.charAt(i)));
      if((len - 5) <= 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;
          }
        }
      }
      } else {
      for(int j = 0; j < 3; j++) {
          for(int l = 0; l <= d; l++) {
            if(l == d) {
              if(l == 3) {
                dp[i][(j + l) % 3][1][0][0] = (dp[i][(j + l) % 3][1][0][0] + dp[i - 1][j][0][0][0]) % MOD;
              } else {
                dp[i][(j + l) % 3][0][0][0] = (dp[i][(j + l) % 3][0][0][0] + dp[i - 1][j][0][0][0]) % MOD;
              }
            } else {
              if(l == 3) {
                dp[i][(j + l) % 3][1][0][1] = (dp[i][(j + l) % 3][1][0][1] + dp[i - 1][j][0][0][0]) % MOD;
              } else {
                dp[i][(j + l) % 3][0][0][1] = (dp[i][(j + l) % 3][0][0][1] + dp[i - 1][j][0][0][0]) % MOD;
              }              
            }
          }
          for(int l = 0; l <= 9; l++) {
            if(l == 3) {
              dp[i][(j + l) % 3][1][0][1] = (dp[i][(j + l) % 3][1][0][1] + dp[i - 1][j][0][0][1]) % MOD;
            } else {
              dp[i][(j + l) % 3][0][0][1] = (dp[i][(j + l) % 3][0][0][1] + dp[i - 1][j][0][0][1]) % MOD;
            }
          }
          for(int l = 0; l <= d; l++) {
            if(l == d) {
              dp[i][(j + l) % 3][1][0][0] = (dp[i][(j + l) % 3][1][0][0] + dp[i - 1][j][1][0][0]) % MOD;
            } else {
              dp[i][(j + l) % 3][1][0][1] = (dp[i][(j + l) % 3][1][0][1] + dp[i - 1][j][1][0][0]) % MOD;
            }
          }
          for(int l = 0; l <= 9; l++) {
            dp[i][(j + l) % 3][1][0][1] = (dp[i][(j + l) % 3][1][0][1] + dp[i - 1][j][1][0][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