結果

問題 No.315 世界のなんとか3.5
ユーザー pythontanukipythontanuki
提出日時 2022-12-27 15:50:38
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 5,581 bytes
コンパイル時間 5,702 ms
コンパイル使用メモリ 275,876 KB
実行使用メモリ 14,008 KB
最終ジャッジ日時 2024-05-01 19:26:15
合計ジャッジ時間 11,265 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
10,652 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 10 ms
6,816 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 10 ms
6,940 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,944 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 2 ms
6,940 KB
testcase_11 AC 2 ms
6,944 KB
testcase_12 AC 516 ms
6,944 KB
testcase_13 AC 494 ms
6,940 KB
testcase_14 AC 979 ms
6,944 KB
testcase_15 AC 986 ms
6,944 KB
testcase_16 TLE -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using namespace std;
using mint = modint1000000007;
using C = complex<double>;
const int mod = 998244353;
const long long LINF = 1001002003004005006;
const int INF = 1001001001;
const double PI = acos(-1);
const double EPS = 1e-10;
const int di[4] = {-1,0,1,0};
const int dj[4] = {0,-1,0,1};
const int dx[8] = {1,1,1,0,0,-1,-1,-1};
const int dy[8] = {1,0,-1,1,-1,1,0,-1};
# define sz(x) (int)(x).size()
# define rsz(x,n) x.resize(n)
# define yosupo(x) {cout << (x) << endl; return 0;}
# define ll long long
# define fi first
# define se second
# define pb push_back
# define pf push_front
# define eb emplace_back
# define ef emplace_front
# define pob pop_back
# define pof pop_front
# define GET_MACRO(_1, _2, _3, NAME, ...) NAME
# define _rep(i, n) _rep2(i, 0, n)
# define _rep2(i, a, b) for(int i = (int)(a); i < (int)(b); i++)
# define rep(...) GET_MACRO(__VA_ARGS__, _rep2, _rep)(__VA_ARGS__)
# define srep(i, a, b) for(int i = a; i <= b; ++i)
# define all(obj) (obj).begin(), (obj).end()
# define rall(obj) (obj).rbegin(), (obj).rend()
inline void YesNo(bool f) { std::cout << (f? "Yes": "No") << std::endl; }
void read() {}
template <typename T, class... U> void read(T &t, U &...u) { cin >> t; read(u...); }
void writeln() { cout << endl; }
template <typename T, class... U, char sep = ' '> void writeln(const T &t, const U &...u) { cout << t; if (sizeof...(u)) cout << sep; writeln(u...); }
# define Pll pair<ll, ll>
# define P pair<int,int>
# define bit(x,i) (((x) >> (i)) & 1)
# define equals(a, b) (fabs((a) - (b)) < EPS) // 誤差を考慮した同値判定
template<class T>bool chmax(T &a, const T &b) { if (a < b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b < a) { a = b; return 1; } return 0; }
template<typename T>istream& operator>>(istream&i,vector<T>&v){rep(j,v.size())i>>v[j];return i;}
template<typename T>string join(vector<T>&v){stringstream s;rep(i,v.size())s<<' '<<v[i];return s.str().substr(1);}
template<typename T>ostream& operator<<(ostream&o,vector<T>&v){if(v.size())o<<join(v);return o;}
template<typename T>ostream& operator<<(ostream&o,vector<vector<T>>&vv){if(vv.size())o<<join(vv);return o;}
template <class T, class U> ostream& operator<<(ostream& os, const pair<T, U>& p) { return os << "P(" << p.first << " " << p.second << ")";}
template<typename T> using vc = vector<T>;
template<typename T> using vv = vc<vc<T>>;
using vi = vc<int>; using vvi = vv<int>;
using vl = vc<ll>; using vvl = vv<ll>;
using vm = vc<mint>; using vvm = vv<mint>;
using vvvm = vc<vvm>; using vvvvm = vc<vvvm>;

mint f(string s, int p) {
    int n = sz(s);
    ll tot = 0;
    int flag = 0;
    ll modP = 0;
    rep(i,n) {
        int nd = s[i]-'0';
        if(nd == 3) flag |= 1;
        tot += nd;
        tot %= 3;
        modP = (modP*10+nd)%p;
    }
    return (mint)((tot%3 == 0 || flag) && modP != 0);
}

mint solve(string s, int p) {
    // modP, mod3, 3があるかないか, 以下フラグ
    // mint dp[p][3][2][2] = {};
    vvvvm dp(p,vvvm(3,vvm(2,vm(2,0))));
    dp[0][0][0][0] = 1;
    int n = sz(s);
    rep(i,n) {
        vvvvm new_dp(p,vvvm(3,vvm(2,vm(2,0))));
        swap(dp,new_dp);
        int nd = s[i]-'0';
        rep(j,2) {
            if(j == 0) {
                rep(modP,p) {
                    rep(mod3,3) {
                        if(nd == 3) {
                            dp[(modP*10+nd)%p][(mod3+nd)%3][1][0] += new_dp[modP][mod3][0][0];
                            dp[(modP*10+nd)%p][(mod3+nd)%3][1][0] += new_dp[modP][mod3][1][0];
                        } else {
                            dp[(modP*10+nd)%p][(mod3+nd)%3][0][0] += new_dp[modP][mod3][0][0];
                            dp[(modP*10+nd)%p][(mod3+nd)%3][1][0] += new_dp[modP][mod3][1][0];
                        }
                    }
                }
            } else {
                rep(k,nd) {
                    rep(modP,p) {
                        rep(mod3,3) {
                            if(k == 3) {
                                dp[(modP*10+k)%p][(mod3+k)%3][1][1] += new_dp[modP][mod3][0][0];
                                dp[(modP*10+k)%p][(mod3+k)%3][1][1] += new_dp[modP][mod3][1][0];
                            } else {
                                dp[(modP*10+k)%p][(mod3+k)%3][0][1] += new_dp[modP][mod3][0][0];
                                dp[(modP*10+k)%p][(mod3+k)%3][1][1] += new_dp[modP][mod3][1][0];
                            }
                        }
                    }
                }
                rep(k,10) {
                    rep(modP,p) {
                        rep(mod3,3) {
                            if(k == 3) {
                                dp[(modP*10+k)%p][(mod3+k)%3][1][1] += new_dp[modP][mod3][0][1];
                                dp[(modP*10+k)%p][(mod3+k)%3][1][1] += new_dp[modP][mod3][1][1];
                            } else {
                                dp[(modP*10+k)%p][(mod3+k)%3][0][1] += new_dp[modP][mod3][0][1];
                                dp[(modP*10+k)%p][(mod3+k)%3][1][1] += new_dp[modP][mod3][1][1];
                            }
                        }
                    }
                }
            }
        }
    }
    mint res = 0;
    rep(modP,p) rep(mod3,3) rep(flag,2) rep(t,2) {
        if((modP != 0) && (mod3 == 0 || flag == 1)) res += dp[modP][mod3][flag][t];
    }
    return res;
}

int main() {
    string s,t; int p;
    read(s,t,p);
    mint res = solve(t,p)-solve(s,p)+f(s,p);
    writeln(res.val());
}
0