結果
問題 | No.315 世界のなんとか3.5 |
ユーザー | takeya_okino |
提出日時 | 2019-07-01 22:49:54 |
言語 | Java21 (openjdk 21) |
結果 |
WA
|
実行時間 | - |
コード長 | 5,620 bytes |
コンパイル時間 | 2,070 ms |
コンパイル使用メモリ | 85,428 KB |
実行使用メモリ | 74,164 KB |
最終ジャッジ日時 | 2024-07-18 08:16:17 |
合計ジャッジ時間 | 15,028 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | WA | - |
testcase_01 | WA | - |
testcase_02 | WA | - |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | WA | - |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | WA | - |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | WA | - |
testcase_19 | WA | - |
testcase_20 | TLE | - |
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 | -- | - |
ソースコード
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[3][2][p][2]; long[][][][] s = new long[3][2][p][2]; int first = Integer.parseInt(String.valueOf(X.charAt(0))); dp[0][0][0][1]++; if(first == 3) { if((len - 5) <= 0) { dp[(first) % 3][1][(first) % p][0]++; } else { dp[(first) % 3][1][0][0]++; } } else { if((len - 5) <= 0) { dp[(first) % 3][0][(first) % p][0]++; } else { dp[(first) % 3][0][0][0]++; } } for(int i = 1; i < first; i++) { if(i == 3) { if((len - 5) <= 0) { dp[i % 3][1][i % p][1]++; } else { dp[i % 3][1][0][1]++; } } else { if((len - 5) <= 0) { dp[i % 3][0][i % p][1]++; } else { dp[i % 3][0][0][1]++; } } } for(int j = 0; j < 3; j++) { for(int k = 0; k < 2; k++) { for(int m = 0; m < p; m++) { for(int l = 0; l < 2; l++) { s[j][k][m][l] = dp[j][k][m][l]; } } } } 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[(j + l) % 3][1][(10 * k + l) % p][0] = (dp[(j + l) % 3][1][(10 * k + l) % p][0] + s[j][0][k][0]) % MOD; } else { dp[(j + l) % 3][0][(10 * k + l) % p][0] = (dp[(j + l) % 3][0][(10 * k + l) % p][0] + s[j][0][k][0]) % MOD; } } else { if(l == 3) { dp[(j + l) % 3][1][(10 * k + l) % p][1] = (dp[(j + l) % 3][1][(10 * k + l) % p][1] + s[j][0][k][0]) % MOD; } else { dp[(j + l) % 3][0][(10 * k + l) % p][1] = (dp[(j + l) % 3][0][(10 * k + l) % p][1] + s[j][0][k][0]) % MOD; } } } for(int l = 0; l <= 9; l++) { if(l == 3) { dp[(j + l) % 3][1][(10 * k + l) % p][1] = (dp[(j + l) % 3][1][(10 * k + l) % p][1] + s[j][0][k][1]) % MOD; } else { dp[(j + l) % 3][0][(10 * k + l) % p][1] = (dp[(j + l) % 3][0][(10 * k + l) % p][1] + s[j][0][k][1]) % MOD; } } for(int l = 0; l <= d; l++) { if(l == d) { dp[(j + l) % 3][1][(10 * k + l) % p][0] = (dp[(j + l) % 3][1][(10 * k + l) % p][0] + s[j][1][k][0]) % MOD; } else { dp[(j + l) % 3][1][(10 * k + l) % p][1] = (dp[(j + l) % 3][1][(10 * k + l) % p][1] + s[j][1][k][0]) % MOD; } } for(int l = 0; l <= 9; l++) { dp[(j + l) % 3][1][(10 * k + l) % p][1] = (dp[(j + l) % 3][1][(10 * k + l) % p][1] + s[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[(j + l) % 3][1][0][0] = (dp[(j + l) % 3][1][0][0] + s[j][0][0][0]) % MOD; } else { dp[(j + l) % 3][0][0][0] = (dp[(j + l) % 3][0][0][0] + s[j][0][0][0]) % MOD; } } else { if(l == 3) { dp[(j + l) % 3][1][0][1] = (dp[(j + l) % 3][1][0][1] + s[j][0][0][0]) % MOD; } else { dp[(j + l) % 3][0][0][1] = (dp[(j + l) % 3][0][0][1] + s[j][0][0][0]) % MOD; } } } for(int l = 0; l <= 9; l++) { if(l == 3) { dp[(j + l) % 3][1][0][1] = (dp[(j + l) % 3][1][0][1] + s[j][0][0][1]) % MOD; } else { dp[(j + l) % 3][0][0][1] = (dp[(j + l) % 3][0][0][1] + s[j][0][0][1]) % MOD; } } for(int l = 0; l <= d; l++) { if(l == d) { dp[(j + l) % 3][1][0][0] = (dp[(j + l) % 3][1][0][0] + s[j][1][0][0]) % MOD; } else { dp[(j + l) % 3][1][0][1] = (dp[(j + l) % 3][1][0][1] + s[j][1][0][0]) % MOD; } } for(int l = 0; l <= 9; l++) { dp[(j + l) % 3][1][0][1] = (dp[(j + l) % 3][1][0][1] + s[j][1][0][1]) % MOD; } } } for(int j = 0; j < 3; j++) { for(int k = 0; k < 2; k++) { for(int m = 0; m < p; m++) { for(int l = 0; l < 2; l++) { s[j][k][m][l] = dp[j][k][m][l]; } } } } } long ret = 0; for(int k = 1; k < p; k++) { for(int l = 0; l < 2; l++) { ret = (ret + dp[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[j][1][k][l]) % MOD; } } } return ret; } }