Submission #49135345

Source Code Expand

#include <atcoder/modint>
#include <bits/stdc++.h>

using namespace std;

using namespace atcoder;
using mint = modint998244353;

int solve(vector<long long> &a) {
    int n = a.size();

    int rootn = sqrt(n);

    vector<mint> dp(n);
    vector<vector<mint>> delta(n, vector<mint>(rootn + 1, 0));
    dp[0] = 1;
    for (int i = 0; i < n; i++) {
        for (int k = 1; k <= rootn; k++) {
            dp[i] += delta[i][k];
        }

        if (a[i] > rootn) {
            for (int j = i + a[i]; j < n; j += a[i]) {
                dp[j] += dp[i];
            }
        } else if (i + a[i] < n) {
            delta[i + a[i]][a[i]] += dp[i];
        }

        for (int k = 1; k <= rootn; k++) {
            if (i + k < n) {
                delta[i + k][k] += delta[i][k];
            }
        }
    }

    mint ans = 0;
    for (int i = 0; i < n; i++) {
        ans += dp[i];
    }
    return ans.val();
}

int main() {
    int n;
    cin >> n;
    vector<long long> a(n);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    cout << solve(a) << endl;
    return 0;
}

Submission Info

Submission Time
Task F - Hop Sugoroku
User adaptatron
Language C++ 20 (gcc 12.2)
Score 525
Code Size 1136 Byte
Status AC
Exec Time 799 ms
Memory 363348 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 525 / 525
Status
AC × 3
AC × 63
Set Name Test Cases
Sample sample_01.txt, sample_02.txt, sample_03.txt
All sample_01.txt, sample_02.txt, sample_03.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt, test_13.txt, test_14.txt, test_15.txt, test_16.txt, test_17.txt, test_18.txt, test_19.txt, test_20.txt, test_21.txt, test_22.txt, test_23.txt, test_24.txt, test_25.txt, test_26.txt, test_27.txt, test_28.txt, test_29.txt, test_30.txt, test_31.txt, test_32.txt, test_33.txt, test_34.txt, test_35.txt, test_36.txt, test_37.txt, test_38.txt, test_39.txt, test_40.txt, test_41.txt, test_42.txt, test_43.txt, test_44.txt, test_45.txt, test_46.txt, test_47.txt, test_48.txt, test_49.txt, test_50.txt, test_51.txt, test_52.txt, test_53.txt, test_54.txt, test_55.txt, test_56.txt, test_57.txt, test_58.txt, test_59.txt, test_60.txt
Case Name Status Exec Time Memory
sample_01.txt AC 1 ms 3488 KiB
sample_02.txt AC 1 ms 3584 KiB
sample_03.txt AC 1 ms 3584 KiB
test_01.txt AC 1 ms 3556 KiB
test_02.txt AC 772 ms 363208 KiB
test_03.txt AC 774 ms 363272 KiB
test_04.txt AC 774 ms 363216 KiB
test_05.txt AC 789 ms 363256 KiB
test_06.txt AC 790 ms 363296 KiB
test_07.txt AC 796 ms 363300 KiB
test_08.txt AC 791 ms 363248 KiB
test_09.txt AC 793 ms 363264 KiB
test_10.txt AC 792 ms 363292 KiB
test_11.txt AC 783 ms 363192 KiB
test_12.txt AC 781 ms 363252 KiB
test_13.txt AC 784 ms 363276 KiB
test_14.txt AC 783 ms 363084 KiB
test_15.txt AC 783 ms 363312 KiB
test_16.txt AC 784 ms 363312 KiB
test_17.txt AC 784 ms 363344 KiB
test_18.txt AC 786 ms 363184 KiB
test_19.txt AC 786 ms 363304 KiB
test_20.txt AC 784 ms 363184 KiB
test_21.txt AC 797 ms 363344 KiB
test_22.txt AC 792 ms 363300 KiB
test_23.txt AC 797 ms 363300 KiB
test_24.txt AC 793 ms 363304 KiB
test_25.txt AC 799 ms 363256 KiB
test_26.txt AC 797 ms 363348 KiB
test_27.txt AC 793 ms 363152 KiB
test_28.txt AC 797 ms 363260 KiB
test_29.txt AC 799 ms 363260 KiB
test_30.txt AC 797 ms 363188 KiB
test_31.txt AC 795 ms 363200 KiB
test_32.txt AC 794 ms 363260 KiB
test_33.txt AC 795 ms 363208 KiB
test_34.txt AC 792 ms 363256 KiB
test_35.txt AC 793 ms 363200 KiB
test_36.txt AC 794 ms 363312 KiB
test_37.txt AC 792 ms 363192 KiB
test_38.txt AC 790 ms 363196 KiB
test_39.txt AC 789 ms 363260 KiB
test_40.txt AC 791 ms 363308 KiB
test_41.txt AC 771 ms 353576 KiB
test_42.txt AC 47 ms 27808 KiB
test_43.txt AC 339 ms 152948 KiB
test_44.txt AC 515 ms 237424 KiB
test_45.txt AC 19 ms 12760 KiB
test_46.txt AC 566 ms 260508 KiB
test_47.txt AC 503 ms 230424 KiB
test_48.txt AC 406 ms 185616 KiB
test_49.txt AC 221 ms 101816 KiB
test_50.txt AC 757 ms 345736 KiB
test_51.txt AC 776 ms 363256 KiB
test_52.txt AC 776 ms 363220 KiB
test_53.txt AC 777 ms 363088 KiB
test_54.txt AC 780 ms 363224 KiB
test_55.txt AC 775 ms 363292 KiB
test_56.txt AC 779 ms 363228 KiB
test_57.txt AC 778 ms 363316 KiB
test_58.txt AC 778 ms 363296 KiB
test_59.txt AC 777 ms 363304 KiB
test_60.txt AC 775 ms 363256 KiB