結果
| 問題 | No.260 世界のなんとか3 |
| コンテスト | |
| ユーザー |
 takeya_okino
|
| 提出日時 | 2019-06-26 23:53:47 |
| 言語 | Java (openjdk 23) |
| 結果 |
RE
|
| 実行時間 | - |
| コード長 | 6,372 bytes |
| 記録 | |
| コンパイル時間 | 2,657 ms |
| コンパイル使用メモリ | 85,952 KB |
| 実行使用メモリ | 48,580 KB |
| 最終ジャッジ日時 | 2024-06-26 07:15:10 |
| 合計ジャッジ時間 | 8,830 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge5 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | RE * 3 |
| other | RE * 27 |
ソースコード
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();
long ans = 0;
if(A.equals("1")) {
ans = func(B);
} else {
ans = func(B) - func(A);
int p1 = 0;
int p2 = 0;
int sum = 0;
for(int i = 0; i < A.length(); i++) {
int a = Integer.parseInt(String.valueOf(A.charAt(i)));
if(a == 3) p1 = 1;
sum += a;
}
if(sum % 3 == 0) p1 = 1;
String b = B;
if(B.length() > 3) b = B.substring(B.length() - 3);
int bn = Integer.parseInt(b);
if((bn % 8) != 0) p2 = 1;
if((p1 == 1) && (p2 == 1)) ans++;
}
System.out.println(ans);
}
public static long func(String X) {
int len = X.length();
long[][][][] dp = new long[len][2][2][2];
long[][][][][] dp2 = new long[len][2][2][2][2];
int first = Integer.parseInt(String.valueOf(X.charAt(0)));
for(int j = 0; j <= first; j++) {
if(j == first) {
if(j == 3) {
dp[0][j % 3][1][0]++;
} else {
dp[0][j % 3][0][0]++;
}
} else {
if(j == 3) {
dp[0][j % 3][1][1]++;
} else {
dp[0][j % 3][0][1]++;
}
}
}
for(int j = 0; j <= first; j++) {
if(j == first) {
if(j == 3) {
dp2[0][j % 3][1][j % 8][0]++;
} else {
dp2[0][j % 3][0][j % 8][0]++;
}
} else {
if(j == 3) {
dp2[0][j % 3][1][j % 8][1]++;
} else {
dp2[0][j % 3][0][j % 8][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 <= d; k++) {
if(k == d) {
if(k == 3) {
dp[i][(j + k) % 3][1][0] = (dp[i][(j + k) % 3][1][0] + dp[i - 1][j][0][0]) % MOD;
} else {
dp[i][(j + k) % 3][0][0] = (dp[i][(j + k) % 3][0][0] + dp[i - 1][j][0][0]) % MOD;
}
} else {
if(k == 3) {
dp[i][(j + k) % 3][1][1] = (dp[i][(j + k) % 3][1][1] + dp[i - 1][j][0][0]) % MOD;
} else {
dp[i][(j + k) % 3][0][1] = (dp[i][(j + k) % 3][0][1] + dp[i - 1][j][0][0]) % MOD;
}
}
}
}
for(int j = 0; j < 3; j++) {
for(int k = 0; k < 10; k++) {
if(k == 3) {
dp[i][(j + k) % 3][1][1] = (dp[i][(j + k) % 3][1][1] + dp[i - 1][j][0][1]) % MOD;
} else {
dp[i][(j + k) % 3][0][1] = (dp[i][(j + k) % 3][0][1] + dp[i - 1][j][0][1]) % MOD;
}
}
}
for(int j = 0; j < 3; j++) {
for(int k = 0; k <= d; k++) {
if(k == d) {
dp[i][(j + k) % 3][1][0] = (dp[i][(j + k) % 3][1][0] + dp[i - 1][j][1][0]) % MOD;
} else {
dp[i][(j + k) % 3][1][1] = (dp[i][(j + k) % 3][1][1] + dp[i - 1][j][1][0]) % MOD;
}
}
}
for(int j = 0; j < 3; j++) {
for(int k = 0; k < 10; k++) {
dp[i][(j + k) % 3][1][1] = (dp[i][(j + k) % 3][1][1] + dp[i - 1][j][1][1]) % MOD;
}
}
}
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 l = 0; l < 8; l++) {
for(int k = 0; k <= d; k++) {
if(k == d) {
if(k == 3) {
dp2[i][(j + k) % 3][1][(10 * l + k) % 8][0] = (dp2[i][(j + k) % 3][1][(10 * l + k) % 8][0] + dp2[i - 1][j][0][(10 * l + k) % 8][0]) % MOD;
} else {
dp2[i][(j + k) % 3][0][(10 * l + k) % 8][0] = (dp2[i][(j + k) % 3][0][(10 * l + k) % 8][0] + dp2[i - 1][j][0][(10 * l + k) % 8][0]) % MOD;
}
} else {
if(k == 3) {
dp2[i][(j + k) % 3][1][(10 * l + k) % 8][1] = (dp2[i][(j + k) % 3][1][(10 * l + k) % 8][1] + dp2[i - 1][j][0][(10 * l + k) % 8][0]) % MOD;
} else {
dp2[i][(j + k) % 3][0][(10 * l + k) % 8][1] = (dp2[i][(j + k) % 3][0][(10 * l + k) % 8][1] + dp2[i - 1][j][0][(10 * l + k) % 8][0]) % MOD;
}
}
}
}
}
for(int j = 0; j < 3; j++) {
for(int l = 0; l < 8; l++) {
for(int k = 0; k < 10; k++) {
if(k == 3) {
dp2[i][(j + k) % 3][1][(10 * l + k) % 8][1] = (dp2[i][(j + k) % 3][1][(10 * l + k) % 8][1] + dp2[i - 1][j][0][(10 * l + k) % 8][1]) % MOD;
} else {
dp2[i][(j + k) % 3][0][(10 * l + k) % 8][1] = (dp2[i][(j + k) % 3][0][(10 * l + k) % 8][1] + dp2[i - 1][j][0][(10 * l + k) % 8][1]) % MOD;
}
}
}
}
for(int j = 0; j < 3; j++) {
for(int l = 0; l < 8; l++) {
for(int k = 0; k <= d; k++) {
if(k == d) {
dp2[i][(j + k) % 3][1][(10 * l + k) % 8][0] = (dp2[i][(j + k) % 3][1][(10 * l + k) % 8][0] + dp2[i - 1][j][1][(10 * l + k) % 8][0]) % MOD;
} else {
dp2[i][(j + k) % 3][1][(10 * l + k) % 8][1] = (dp2[i][(j + k) % 3][1][(10 * l + k) % 8][1] + dp2[i - 1][j][1][(10 * l + k) % 8][0]) % MOD;
}
}
}
}
for(int j = 0; j < 3; j++) {
for(int l = 0; l < 8; l++) {
for(int k = 0; k < 10; k++) {
dp2[i][(j + k) % 3][1][(10 * l + k) % 8][1] = (dp2[i][(j + k) % 3][1][(10 * l + k) % 8][1] + dp2[i - 1][j][1][(10 * l + k) % 8][1]) % MOD;
}
}
}
}
long t1 = 0;
long t2 = 0;
for(int k = 0; k < 2; k++) {
if(k == 0) {
for(int l = 0; l < 2; l++) {
t1 = (t1 + dp[len - 1][0][k][l]) % MOD;
}
} else {
for(int l = 0; l < 2; l++) {
for(int j = 0; j < 3; j++) {
t1 = (t1 + dp[len - 1][j][k][l]) % MOD;
}
}
}
}
for(int k = 0; k < 2; k++) {
if(k == 0) {
for(int l = 0; l < 2; l++) {
t2 = (t2 + dp2[len - 1][0][k][0][l]) % MOD;
}
} else {
for(int l = 0; l < 2; l++) {
for(int j = 0; j < 3; j++) {
t2 = (t2 + dp2[len - 1][j][k][0][l]) % MOD;
}
}
}
}
long ans = (t1 - t2 + MOD) % MOD;
return ans;
}
}