제출 #1051415

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1051415		deera	Trains (BOI24_trains)	C++14	100 / 100	266 ms	398708 KiB

Main

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

const ll MOD = 1e9+7;

int main() {
    ll n; cin >> n;
    ll m = n + 100;
    
    const ll p = 500; // (ll)sqrt(n);

    vector<ll> dp(m);
    vector<ll> d(m), x(m);
    vector<vector<ll>> rest(m, vector<ll>(p, 0));

    for (ll i = 1; i <= n; i++) cin >> d[i] >> x[i];
    for (ll i = 0; i < dp.size(); i++) dp[i] = 1;

    for (ll i = n; i >= 1; i--) {
        ll step = d[i];
        ll next = i + step;
        ll maxd = i + step * (x[i] + 1);

        if (d[i] != 0) { // it's not broken
            if (step >= p) {
                for (ll j = next; j <= n && j <= maxd - step; j+=step) {
                    dp[i] = (dp[j] + dp[i]) % MOD;
                }
            } else {
                if (next <= n) dp[i] += rest[next][step];
                if (maxd <= n) dp[i] -= rest[maxd][step];
                dp[i] %= MOD; // modding here to avoid moddings negatives
            }
        }

        for (int step = 1; step < p; step++) {
            ll next = i + step;

            if (next <= n)  
                rest[i][step] = (dp[i] + rest[next][step]) % MOD;
            else
                rest[i][step] = (dp[i]) % MOD;
        }
    }

    dp[1] += MOD;
    cout << dp[1] % MOD << endl;
    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

Main.cpp: In function 'int main()':
Main.cpp:19:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   19 |     for (ll i = 0; i < dp.size(); i++) dp[i] = 1;
      |                    ~~^~~~~~~~~~~

• Subtask 1: 8 / 8
• Subtask 2: 13 / 13
• Subtask 3: 16 / 16
• Subtask 4: 34 / 34
• Subtask 5: 29 / 29