結果

問題 No.315 世界のなんとか3.5
ユーザー kimiyukikimiyuki
提出日時 2017-01-08 18:23:05
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,125 bytes
コンパイル時間 937 ms
コンパイル使用メモリ 80,636 KB
実行使用メモリ 17,480 KB
最終ジャッジ日時 2024-12-17 18:07:40
合計ジャッジ時間 65,813 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17 TLE * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <array>
#define repeat(i,n) for (int i = 0; (i) < int(n); ++(i))
#define whole(f,x,...) ([&](decltype((x)) whole) { return (f)(begin(whole), end(whole), ## __VA_ARGS__); })(x)
using ll = long long;
using namespace std;
template <typename X, typename T> auto vectors(X x, T a) { return vector<T>(x, a); }
template <typename X, typename Y, typename Z, typename... Zs> auto vectors(X x, Y y, Z z, Zs... zs) { auto cont = vectors(y, z, zs...); return vector<decltype(cont)>(x, cont); }
vector<vector<ll> > strmodin(string const & s, int a, vector<int> const & b, ll m) {
    int pow_b = 1 << b.size();
    array<int,10> table = {};
    repeat (d,10) {
        auto it = whole(find, b, d);
        if (it != b.end()) table[d] |= 1 << (it - b.begin());
    }
    auto cur = vectors(a, pow_b, ll());
    auto prv = vectors(a, pow_b, ll());
    int bound_i = 0, bound_j = 0;
    for (char c : s) {
        c -= '0';
        cur.swap(prv);
        repeat (d, c) cur[(bound_i * 10 + d) % a][bound_j | table[d]] += 1;
        bound_i = (bound_i * 10 + c) % a;
        bound_j |= table[c];
        repeat (i,a) repeat (j,pow_b) {
            repeat (d,10) {
                ll & it = cur[(i * 10 + d) % a][j | table[d]];
                it = (it + prv[i][j]) % m;
            }
            prv[i][j] = 0;
        }
    }
    cur[bound_i][bound_j] += 1;
    cur[bound_i][bound_j] %= m;
    return cur;
}
const int mod = 1e9+7;
ll f(string const & s, int p) {
    auto dp = strmodin(s, 3 * p, { 3 }, mod);
    ll cnt = 0;
    repeat (i,3*p) repeat (j,2) {
        if ((i % 3 == 0 or j) and i % p != 0) {
            cnt += dp[i][j];
            cnt %= mod;
        }
    }
    return cnt;
}
bool g(string const & s, int p) {
    auto modstr = [&](ll m) {
        ll q = 0;
        for (char c : s) q = (q * 10 + (c - '0')) % m;
        return q;
    };
    return (modstr(3) == 0 or whole(count, s, '3')) and modstr(p) != 0;
}
int main() {
    string a, b; int p; cin >> a >> b >> p;
    cout << ((f(b, p) - f(a, p) + g(a, p)) % mod + mod) % mod << endl;
    return 0;
}
0