結果

問題 No.260 世界のなんとか3
コンテスト
ユーザー siman
提出日時 2023-07-03 05:56:49
言語 C++17(clang)
(17.0.6 + boost 1.87.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

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