結果

問題 No.260 世界のなんとか3
コンテスト
ユーザー hos.lyric
提出日時 2019-01-21 10:26:30
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 144 ms / 2,000 ms
コード長 2,621 bytes
コンパイル時間 1,002 ms
コンパイル使用メモリ 121,948 KB
実行使用メモリ 17,408 KB
最終ジャッジ日時 2024-06-13 03:28:21
合計ジャッジ時間 5,191 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.conv, std.stdio, std.string;
import std.algorithm, std.array, std.bigint, std.container, std.math, std.numeric, std.range, std.regex, std.typecons;
import core.bitop;

class EOFException : Throwable { this() { super("EOF"); } }
string[] tokens;
string readToken() { for (; tokens.empty; ) { if (stdin.eof) { throw new EOFException; } tokens = readln.split; } auto token = tokens.front; tokens.popFront; return token; }
int readInt() { return readToken.to!int; }
long readLong() { return readToken.to!long; }
real readReal() { return readToken.to!real; }

void chmin(T)(ref T t, in T f) { if (t > f) t = f; }
void chmax(T)(ref T t, in T f) { if (t < f) t = f; }

int binarySearch(T)(in T[] as, in bool delegate(T) test) { int low = -1, upp = cast(int)(as.length); for (; low + 1 < upp; ) { int mid = (low + upp) >> 1; (test(as[mid]) ? low : upp) = mid; } return upp; }
int lowerBound(T)(in T[] as, in T val) { return as.binarySearch((T a) => (a < val)); }
int upperBound(T)(in T[] as, in T val) { return as.binarySearch((T a) => (a <= val)); }


enum MO = 10L^^9 + 7;
enum M = 24;
enum LIM = 10^^4 + 10;

int[] ten;
long[][][] dp;

int N;
string A, B;

long calc(int i, int x, int y, bool diffA, bool diffB) {
  long ret;
  if (i == N) {
    ret += ((x % 3 == 0 || y) && (x % 8 != 0)) ? 1 : 0;
  } else if (diffA && diffB) {
    foreach (dx; 0 .. M) foreach (dy; 0 .. 2) {
      const xx = (x * ten[N - i] + dx) % M;
      const yy = y | dy;
      if ((xx % 3 == 0 || yy) && (xx % 8 != 0)) {
        ret += dp[N - i][dx][dy];
      }
    }
  } else {
    foreach (d; (diffA ? 0 : (A[i] - '0')) .. (diffB ? 9 : (B[i] - '0')) + 1) {
      ret += calc(i + 1, (x * 10 + d) % M, (y || d == 3) ? 1 : 0, (diffA || (d != A[i] - '0')), (diffB || (d != B[i] - '0')));
    }
  }
  ret %= MO;
  debug {
    writefln("calc %s %s %s %s %s: %s", i, x, y, diffA, diffB, ret);
  }
  return ret;
}

void main() {
  ten = new int[LIM];
  ten[0] = 1;
  foreach (i; 1 .. LIM) {
    ten[i] = (ten[i - 1] * 10) % M;
  }
  
  dp = new long[][][](LIM, M, 2);
  dp[0][0][0] = 1;
  foreach (i; 1 .. LIM) {
    foreach (x; 0 .. M) foreach (y; 0 .. 2) {
      foreach (d; 0 .. 10) {
        (dp[i][(x * 10 + d) % M][(y || d == 3) ? 1 : 0] += dp[i - 1][x][y]) %= MO;
      }
    }
  }
  
  try {
    for (; ; ) {
      A = readToken();
      B = readToken();
      N = cast(int)(B.length);
      A = iota(N - A.length).map!(_ => '0').array.to!string ~ A;
      debug {
        writeln("A = ", A);
        writeln("B = ", B);
      }
      const ans = calc(0, 0, 0, false, false);
      writeln(ans);
    }
  } catch (EOFException e) {
  }
}
0