結果

問題 No.260 世界のなんとか3
ユーザー nomi
提出日時 2019-02-16 17:11:13
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 76 ms / 2,000 ms
コード長 1,706 bytes
コンパイル時間 1,591 ms
コンパイル使用メモリ 170,780 KB
実行使用メモリ 21,824 KB
最終ジャッジ日時 2024-09-22 14:32:52
合計ジャッジ時間 3,786 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define int long long
#define rep(i, n) for (int (i) = (0); (i) < (n); ++(i))

template<class T>
void Add(T &a, const T &b, const T &mod=1000000007) {
  int val = ((a % mod) + (b % mod)) % mod;
  if (val < 0) { val += mod; }
  a = val;
}

// ------------------------------------------------------------------------------------------

string A, B;
int dp[11111][2][2][2][3][8]; // dp[桁][Aを超えるか][B未満か][3を含むか][3で割ったあまり][8で割ったあまり] = 総数

int rec(int digit, int over, int smaller, int three, int mod3, int mod8) {
  // 桁数超えたら終了
  if (digit >= B.size()) {
    return (three || mod3 == 0) && (mod8 != 0);
  }

  // メモされてるやつだったら返す
  if (dp[digit][over][smaller][three][mod3][mod8] != -1) {
    return dp[digit][over][smaller][three][mod3][mod8];
  }

  int ue = (smaller ? 9 : B[digit] - '0'); // 下限
  int shita = (over ? 0 : A[digit] - '0'); // 上限

  int ret = 0;
  for (int num = shita; num <= ue; num++) {
    int t = rec(digit+1, over || (num > shita), smaller || (num < ue), three || (num == 3), (mod3+num)%3, (10*mod8+num)%8);
    Add(ret, t);
  }

  return dp[digit][over][smaller][three][mod3][mod8] = ret;
}

signed main() {
  cin >> A >> B;
  // Aの桁数が小さい時、Aの上位桁を0で埋める
  int diff = B.size() - A.size();
  string zero = "";
  rep (i, diff) {
    zero += "0";
  }
  A = zero + A;

  // DPテーブル初期化
  rep (i, 11111) rep (j, 2) rep (k, 2) rep (l, 2) rep (m, 3) rep (n, 8) dp[i][j][k][l][m][n] = -1;

  // メモ化再帰に投げる
  int ans = rec(0, 0, 0, 0, 0, 0);
  cout << ans << endl;

  return 0;
}
0