結果

問題 No.260 世界のなんとか3
ユーザー simansiman
提出日時 2023-07-03 05:56:49
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
AC  
実行時間 75 ms / 2,000 ms
コード長 3,861 bytes
コンパイル時間 1,128 ms
コンパイル使用メモリ 143,092 KB
実行使用メモリ 11,068 KB
最終ジャッジ日時 2024-07-17 02:44:43
合計ジャッジ時間 3,008 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 60 ms
10,944 KB
testcase_04 AC 61 ms
11,068 KB
testcase_05 AC 13 ms
6,944 KB
testcase_06 AC 10 ms
6,944 KB
testcase_07 AC 42 ms
9,844 KB
testcase_08 AC 30 ms
9,960 KB
testcase_09 AC 18 ms
6,940 KB
testcase_10 AC 47 ms
9,196 KB
testcase_11 AC 44 ms
10,416 KB
testcase_12 AC 27 ms
8,552 KB
testcase_13 AC 9 ms
6,944 KB
testcase_14 AC 41 ms
9,984 KB
testcase_15 AC 12 ms
6,944 KB
testcase_16 AC 35 ms
8,688 KB
testcase_17 AC 28 ms
7,144 KB
testcase_18 AC 27 ms
6,940 KB
testcase_19 AC 36 ms
9,964 KB
testcase_20 AC 25 ms
7,528 KB
testcase_21 AC 23 ms
7,788 KB
testcase_22 AC 41 ms
8,812 KB
testcase_23 AC 6 ms
6,940 KB
testcase_24 AC 32 ms
9,308 KB
testcase_25 AC 39 ms
10,928 KB
testcase_26 AC 25 ms
10,860 KB
testcase_27 AC 2 ms
6,944 KB
testcase_28 AC 61 ms
11,000 KB
testcase_29 AC 75 ms
11,068 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cassert>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <climits>
#include <map>
#include <queue>
#include <set>
#include <cstring>
#include <vector>

using namespace std;
typedef long long ll;

const ll MOD = 1000000007;

ll mod_pow(ll x, ll n, ll mod = MOD) {
  ll res = 1;

  while (n > 0) {
    if (n & 1) {
      res = res * x % mod;
    }

    x = x * x % mod;
    n >>= 1;
  }

  return res;
}

ll f(string S) {
  int len = S.size();
  ll dp1[len][2][3][8];
  ll dp2[len][2][3][8];
  memset(dp1, 0, sizeof(dp1));
  memset(dp2, 0, sizeof(dp2));

  for (int i = 0; i < len; ++i) {
    ll base3 = mod_pow(10, len - i - 1, 3);
    ll base8 = mod_pow(10, len - i - 1, 8);
    int d = S[i] - '0';

    if (i == 0) {
      for (int v = d; v >= 1; --v) {
        int m3 = (v * base3) % 3;
        int m8 = (v * base8) % 8;

        if (v == d) {
          if (v != 3) {
            dp1[i][0][m3][m8] += 1;
          } else {
            dp1[i][1][m3][m8] += 1;
          }
        } else {
          if (v != 3) {
            dp2[i][0][m3][m8] += 1;
          } else {
            dp2[i][1][m3][m8] += 1;
          }
        }
      }
    } else {
      for (int u = 1; u <= 9; ++u) {
        int n_m3 = (u * base3) % 3;
        int n_m8 = (u * base8) % 8;

        if (u != 3) {
          dp2[i][0][n_m3][n_m8] += 1;
        } else {
          dp2[i][1][n_m3][n_m8] += 1;
        }
      }

      for (int b_m3 = 0; b_m3 < 3; ++b_m3) {
        for (int b_m8 = 0; b_m8 < 8; ++b_m8) {
          for (int has_3 = 0; has_3 < 2; ++has_3) {
            {
              int n_m3 = (d * base3 + b_m3) % 3;
              int n_m8 = (d * base8 + b_m8) % 8;

              if (d != 3) {
                dp1[i][has_3][n_m3][n_m8] += dp1[i - 1][has_3][b_m3][b_m8];
              } else {
                dp1[i][1][n_m3][n_m8] += dp1[i - 1][has_3][b_m3][b_m8];
              }
            }

            for (int u = 0; u < d; ++u) {
              int n_m3 = (u * base3 + b_m3) % 3;
              int n_m8 = (u * base8 + b_m8) % 8;

              if (u != 3) {
                dp2[i][has_3][n_m3][n_m8] += dp1[i - 1][has_3][b_m3][b_m8];
                dp2[i][has_3][n_m3][n_m8] %= MOD;
              } else {
                dp2[i][1][n_m3][n_m8] += dp1[i - 1][has_3][b_m3][b_m8];
                dp2[i][1][n_m3][n_m8] %= MOD;
              }
            }

            for (int u = 0; u <= 9; ++u) {
              int n_m3 = (u * base3 + b_m3) % 3;
              int n_m8 = (u * base8 + b_m8) % 8;

              if (u != 3) {
                dp2[i][has_3][n_m3][n_m8] += dp2[i - 1][has_3][b_m3][b_m8];
                dp2[i][has_3][n_m3][n_m8] %= MOD;
              } else {
                dp2[i][1][n_m3][n_m8] += dp2[i - 1][has_3][b_m3][b_m8];
                dp2[i][1][n_m3][n_m8] %= MOD;
              }
            }
          }
        }
      }
    }
  }

  ll cnt = 0;

  for (int m3 = 0; m3 < 3; ++m3) {
    for (int m8 = 0; m8 < 8; ++m8) {
      for (int has_3 = 0; has_3 < 2; ++has_3) {
        if ((m3 == 0 || has_3) && m8 != 0) {
          cnt += dp1[len - 1][has_3][m3][m8];
          cnt %= MOD;
          cnt += dp2[len - 1][has_3][m3][m8];
          cnt %= MOD;
        }
      }
    }
  }

  return cnt;
}

string str_dec(string str) {
  if (str == "1") {
    return "0";
  }

  reverse(str.begin(), str.end());
  int len = str.size();

  for (int i = 0; i < len; ++i) {
    if (str[i] != '0') {
      str[i]--;
      break;
    }
    str[i] = '9';
  }

  if (str.back() == '0') {
    str.resize(len - 1);
  }

  reverse(str.begin(), str.end());
  return str;
}

int main() {
  string A, B;
  cin >> A >> B;

  ll cnt1 = f(B);
  ll cnt2 = f(str_dec(A));

  // cerr << f("100000000") << endl;
  // fprintf(stderr, "(%lld, %lld)\n", cnt1, cnt2);
  cout << (cnt1 - cnt2 + MOD) % MOD << endl;

  return 0;
}
0