結果

問題 No.260 世界のなんとか3
ユーザー takeya_okinotakeya_okino
提出日時 2019-06-27 00:04:53
言語 Java21
(openjdk 21)
結果
MLE  
実行時間 -
コード長 6,238 bytes
コンパイル時間 2,588 ms
コンパイル使用メモリ 85,732 KB
実行使用メモリ 93,688 KB
最終ジャッジ日時 2024-06-26 07:30:51
合計ジャッジ時間 23,467 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 130 ms
40,816 KB
testcase_01 AC 130 ms
41,380 KB
testcase_02 AC 130 ms
40,880 KB
testcase_03 MLE -
testcase_04 MLE -
testcase_05 MLE -
testcase_06 WA -
testcase_07 MLE -
testcase_08 MLE -
testcase_09 MLE -
testcase_10 MLE -
testcase_11 MLE -
testcase_12 MLE -
testcase_13 MLE -
testcase_14 MLE -
testcase_15 MLE -
testcase_16 MLE -
testcase_17 MLE -
testcase_18 MLE -
testcase_19 MLE -
testcase_20 MLE -
testcase_21 MLE -
testcase_22 MLE -
testcase_23 WA -
testcase_24 MLE -
testcase_25 MLE -
testcase_26 MLE -
testcase_27 AC 134 ms
41,124 KB
testcase_28 MLE -
testcase_29 MLE -
権限があれば一括ダウンロードができます

ソースコード

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();
    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][3][2][2];
    long[][][][][] dp2 = new long[len][3][2][8][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][l][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][l][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][l][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][l][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][l][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][l][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][l][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][l][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][l][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;
  }
}
0