結果

問題 No.315 世界のなんとか3.5
ユーザー pekempey
提出日時 2015-12-08 12:02:15
言語 C++11(廃止可能性あり)
(gcc 13.3.0)
結果
AC  
実行時間 388 ms / 2,000 ms
コード長 1,793 bytes
コンパイル時間 1,493 ms
コンパイル使用メモリ 163,224 KB
実行使用メモリ 23,472 KB
最終ジャッジ日時 2024-09-14 19:53:00
合計ジャッジ時間 6,842 ms
ジャッジサーバーID
(参考情報)
judge5 / judge6
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define GET_MACRO(a, b, c, NAME, ...) NAME
#define rep(...) GET_MACRO(__VA_ARGS__, rep3, rep2)(__VA_ARGS__)
#define rep2(i, a) rep3 (i, 0, a)
#define rep3(i, a, b) for (int i = (a); i < (b); i++)
#define repr(...) GET_MACRO(__VA_ARGS__, repr3, repr2)(__VA_ARGS__)
#define repr2(i, a) repr3 (i, 0, a)
#define repr3(i, a, b) for (int i = (b) - 1; i >= (a); i--)
template<class T1, class T2> inline bool chmin(T1 &a, T2 b) { return b < a && (a = b, true); }
template<class T1, class T2> inline bool chmax(T1 &a, T2 b) { return a < b && (a = b, true); }
using namespace std;
typedef long long ll;

const ll mod = 1e9 + 7;
ll dp[202020][2][3][2]; // pos, less, mod 3, three
ll dp2[8][2][3][2][800]; // pos, less, mod 3, three, mod P

ll f(string N, int P, bool less) {
	memset(dp, 0, sizeof(dp));
	memset(dp2, 0, sizeof(dp2));

	int s = max<int>(0, N.length() - 6);
	int t = N.length() - s;

	dp[0][0][0][0] = 1;
	rep (i, s) rep (j, 2) rep (k, 3) rep (l, 2) {
		int lim = j ? 9 : N[i] - '0';
		rep (x, lim + 1) {
			(dp[i + 1][j || x < lim][(k + x) % 3][l || x == 3] += dp[i][j][k][l]) %= mod;
		}
	}
	rep (j, 2) rep (k, 3) rep (l, 2) {
		dp2[0][j][k][l][0] = dp[s][j][k][l];
	}

	rep (i, t) rep (j, 2) rep (k, 3) rep (l, 2) rep (m, P) {
		int lim = j ? 9  : N[s + i] - '0';
		rep (x, lim + 1) {
			(dp2[i + 1][j || x < lim][(k + x) % 3][l || x == 3][(m * 10 + x) % P] += dp2[i][j][k][l][m]) %= mod;
		}
	}

	ll res = 0;
	rep (j, 2) rep (k, 3) rep (l, 2) rep (m, P) {
		if (less && j == 0) continue;
		if ((k == 0 || l) && m != 0) {
			(res += dp2[t][j][k][l][m]) %= mod;
		}
	}
	return res;
}

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

	ll ans = f(B, P, false);
	ans -= f(A, P, true);
	ans %= mod; ans += mod; ans %= mod;
	cout << ans << endl;
	return 0;
}
0