結果

問題 No.315 世界のなんとか3.5
ユーザー mkawa2mkawa2
提出日時 2024-09-23 22:35:13
言語 C++23(gcc13)
(gcc 13.2.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,444 bytes
コンパイル時間 6,118 ms
コンパイル使用メモリ 99,504 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-23 22:39:54
合計ジャッジ時間 21,521 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 24 ms
6,944 KB
testcase_08 AC 2 ms
6,944 KB
testcase_09 AC 1 ms
6,944 KB
testcase_10 AC 2 ms
6,940 KB
testcase_11 AC 2 ms
6,944 KB
testcase_12 AC 190 ms
6,940 KB
testcase_13 AC 191 ms
6,944 KB
testcase_14 AC 379 ms
6,940 KB
testcase_15 AC 381 ms
6,944 KB
testcase_16 AC 378 ms
6,940 KB
testcase_17 AC 378 ms
6,940 KB
testcase_18 AC 192 ms
6,940 KB
testcase_19 AC 219 ms
6,944 KB
testcase_20 AC 194 ms
6,940 KB
testcase_21 AC 194 ms
6,944 KB
testcase_22 WA -
testcase_23 AC 379 ms
6,944 KB
testcase_24 AC 377 ms
6,940 KB
testcase_25 WA -
testcase_26 AC 379 ms
6,940 KB
testcase_27 AC 378 ms
6,940 KB
testcase_28 AC 377 ms
6,940 KB
testcase_29 AC 777 ms
6,940 KB
testcase_30 AC 754 ms
6,940 KB
testcase_31 AC 752 ms
6,944 KB
testcase_32 AC 782 ms
6,944 KB
testcase_33 AC 381 ms
6,944 KB
testcase_34 AC 748 ms
6,944 KB
testcase_35 AC 757 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
using namespace std;

const long long inf = -1 - (-1LL << 62);
const int md = 1e9 + 7;

long long digitdp(vector<int>& aa, bool eq, int k) {
    long long res = 0;
    vector<vector<long long>> dp(2, vector<long long>(24, 0));
    int jf = 0, jm = 0;

    for (size_t i = 0; i < aa.size(); ++i) {
        vector<vector<long long>> ndp(2, vector<long long>(24, 0));
        
        // Transition from less than confirmed
        for (int f = 0; f < 2; ++f) {
            for (int m = 0; m < 24; ++m) {
                long long pre = dp[f][m] % md;
                if (pre == 0) continue;
                for (int d = 0; d < 10; ++d) {
                    int nf = f | (d == 3);
                    int nm = (m * 10 + d) % 24;
                    ndp[nf][nm] += pre;
                    ndp[nf][nm] %=md;
                }
            }
        }
        
        // Transition from exactly
        for (int d = (i == 0); d < aa[i]; ++d) {
            int nf = jf | (d == 3);
            int nm = (jm * 10 + d) % 24;
            ndp[nf][nm] += 1;
        }
        
        // Start from this digit
        if (i) {
            for (int d = 1; d < 10; ++d) {
                ndp[d == 3][d] += 1;
            }
        }
        
        // Update exactly
        jf |= (aa[i] == 3);
        jm = (jm * 10 + aa[i]) % 24;

        dp = ndp;

        if (i == aa.size() - k) {
            if (eq && (jf || jm % 3 == 0) && jm % 8 == 0) res -= 1;
            for (int f = 0; f < 2; ++f) {
                for (int m = 0; m < 24; ++m) {
                    if ((f || m % 3 == 0) && m % 8 == 0) {
                        res -= dp[f][m];
                        res %= md;
                    }
                }
            }
        }
    }

    if (eq && (jf || jm % 3 == 0)) res += 1;
    for (int f = 0; f < 2; ++f) {
        for (int m = 0; m < 24; ++m) {
            if (f || m % 3 == 0) {
                res += dp[f][m];
                res %= md;
            }
        }
    }

    return res;
}

int main() {
    string a, b, p;
    cin >> a >> b >> p;

    vector<int> aa(a.size()), bb(b.size());
    for (size_t i = 0; i < a.size(); ++i) {
        aa[i] = a[i] - '0';
    }
    for (size_t i = 0; i < b.size(); ++i) {
        bb[i] = b[i] - '0';
    }
    int k = p.size();
    long long ans = digitdp(bb, true, k) - digitdp(aa, false, k);
    cout << (ans % md + md) % md << endl;

    return 0;
}

0