QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#250234#4590. Happy Travellingwarner1129WA 14ms17204kbC++202.9kb2023-11-12 23:35:152023-11-12 23:35:17

Judging History

你现在查看的是最新测评结果

  • [2023-11-12 23:35:17]
  • 评测
  • 测评结果:WA
  • 用时:14ms
  • 内存:17204kb
  • [2023-11-12 23:35:15]
  • 提交

answer

#include <bits/stdc++.h>

using namespace std;

#ifdef LOCAL
template<class... T> void dbg(T... x) { char e{}; ((cerr << e << x, e = ' '), ...); }
template<class T> void org(T l, T r) { while (l != r) cerr << ' ' << *l++; cerr << '\n'; }
#define debug(x...) dbg(#x, '=', x, '\n')
#define olist(x...) dbg(#x, '='), org(x)
#else
#define debug(...) ((void)0)
#define olist(...) ((void)0)
#endif
#define all(v) (v).begin(), (v).end()
#define rall(v) (v).rbegin(), (v).rend()
#define ff first
#define ss second

using i64 = long long;
using u64 = unsigned long long;
using i128 = __int128;
using u128 = unsigned __int128;

template<class T>
inline constexpr T inf = numeric_limits<T>::max() / 2;
constexpr int mod = 998244353;

template<class T> bool chmin(T &a, T b) { return (b < a and (a = b, true)); }
template<class T> bool chmax(T &a, T b) { return (a < b and (a = b, true)); }
template<class... T> int add(T... x) { int t{}; return (((t += x) %= mod), ...), t; }
template<class... T> int mul(T... x) { i64 t{1}; return (((t *= x) %= mod), ...), t; }

void solve() {
    int n, K, D;
    cin >> n >> K >> D;

    vector<int> H(n);
    vector<int> T(n - 1);
    for (int &x : H) cin >> x;
    for (int &x : T) cin >> x;

    if (K * K <= 1e5) {
        vector<vector<int>> stk(K);
        vector<i64> dp(n, -inf<i64>);
        dp[n - 1] = H[n - 1];
        stk[(n - 1) % K].push_back(n - 1);
        for (int i = n - 2; i >= 0; i--) {
            for (int j = 0; j < K; j++) {
                auto it = lower_bound(all(stk[j]), i + T[i], greater<int>{});
                if (it != stk[j].end()) {
                    chmax(dp[i], dp[*it] - (*it - i) / K * D + H[i]);
                }
            }
            int s = i % K;
            while (!stk[s].empty() and dp[stk[s].back()] - stk[s].back() / K * D < dp[i] - i / K * D) {
                stk[s].pop_back();
            }
            stk[s].push_back(i);
        }
        cout << dp[0] << '\n';
    } else {
        const int lgK = __lg(K);
        vector st(lgK + 1, vector<i64>(n, -inf<i64>));
        vector<i64> dp(n);
        dp[n - 1] = H[n - 1];

        auto upd = [&](int idx) -> void {
            st[0][idx] = dp[idx];
            for (int i = 0; i + 1 <= lgK and idx + (2 << i) <= n; i++) {
                st[i + 1][idx] = max(st[i][idx], st[i][idx + (1 << i)]);
            }
        };

        auto qry = [&](int l, int r) -> i64 {
            int s = __lg(r - l);
            return max(st[s][l], st[s][r - (1 << s)]);
        };

        upd(n - 1);
        for (int i = n - 2; i >= 0; i--) {
            for (int j = i; j <= i + T[i]; j += K) {
                chmax(dp[i], qry(max(j, i + 1), min(j + K, i + T[i] + 1)) - (j - i) / K * D + H[i]);
            }
            upd(i);
        }
        cout << dp[0] << '\n';
    }
    
}

signed main() {
    cin.tie(0)->sync_with_stdio(false);
    cin.exceptions(cin.failbit);
    
    int T = 1;
    // cin >> T;
    while (T--) {
        solve();
    }
    
    return 0;
}


Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3400kb

input:

6 2 1
8 -7 -8 9 0 2
5 3 3 2 1

output:

18

result:

ok single line: '18'

Test #2:

score: 0
Accepted
time: 0ms
memory: 3428kb

input:

8 8 8
10 -5 -5 -5 -5 -5 -5 10
5 2 5 3 2 1 1

output:

15

result:

ok single line: '15'

Test #3:

score: 0
Accepted
time: 0ms
memory: 3400kb

input:

13 2 2
-5 -4 -4 -1 7 -6 -5 -4 -3 -2 -1 5 -7
3 10 9 8 7 6 5 4 3 2 1 1

output:

-9

result:

ok single line: '-9'

Test #4:

score: 0
Accepted
time: 0ms
memory: 3392kb

input:

2 1 0
-10000 10000
1

output:

0

result:

ok single line: '0'

Test #5:

score: 0
Accepted
time: 14ms
memory: 4568kb

input:

98987 4 3
-8225 -8961 -5537 -5621 -8143 -5214 -5538 -6912 -6601 -8839 -7872 -7867 -9553 -9793 -7333 -7360 -5820 -7459 -8824 -9716 -9757 -5846 -5300 -5912 -7953 -8360 -7609 -5937 -5525 -9748 -7326 -8311 -9979 -9292 -8542 -7589 -7939 -5914 -7985 -9999 -9212 -8274 -8084 -6620 -5991 -7826 -6327 -5228 -6...

output:

-84108

result:

ok single line: '-84108'

Test #6:

score: 0
Accepted
time: 13ms
memory: 4560kb

input:

98467 2 3
-5677 -9080 -6665 -5838 -5755 -8938 -6286 -5160 -7147 -8370 -8214 -6088 -9763 -5183 -7123 -7264 -5298 -8855 -6381 -6592 -9216 -8429 -9598 -7443 -7393 -8712 -5545 -6778 -6010 -5717 -9102 -7968 -6140 -9592 -7917 -5217 -5015 -7798 -9339 -5678 -7073 -7607 -7961 -6185 -9941 -6421 -8779 -5388 -8...

output:

-150169

result:

ok single line: '-150169'

Test #7:

score: 0
Accepted
time: 12ms
memory: 4624kb

input:

96173 2 1
-6463 -9099 -6269 -6169 -8273 -9839 -9929 -5447 -5908 -6884 -6908 -8359 -6477 -9414 -9207 -8180 -6264 -9293 -8981 -6557 -9260 -9700 -6785 -7121 -8382 -9712 -5178 -5821 -9107 -9004 -7472 -9306 -9311 -7160 -7965 -5394 -8048 -7415 -5233 -7746 -7390 -5298 -7721 -7915 -9646 -5371 -5712 -6234 -5...

output:

-45780

result:

ok single line: '-45780'

Test #8:

score: 0
Accepted
time: 8ms
memory: 4616kb

input:

96905 1 1
-7933 -5685 -6201 -5415 -7294 -9904 -8968 -8433 -6287 -6727 -5933 -5418 -8953 -6743 -7057 -7087 -7244 -5302 -5130 -8934 -5053 -9811 -8695 -5989 -7785 -5991 -8351 -9463 -7037 -8867 -8956 -8194 -5272 -6270 -7737 -7793 -8851 -5839 -5178 -7009 -8018 -7375 -6091 -9226 -7117 -9522 -9248 -6036 -8...

output:

-72336

result:

ok single line: '-72336'

Test #9:

score: 0
Accepted
time: 9ms
memory: 4652kb

input:

97070 2 1
-6843 -9736 -7145 -9801 -6733 -7807 -8835 -5776 -6971 -7101 -8428 -7429 -9250 -9405 -6152 -7904 -5761 -8347 -6597 -8875 -8741 -7759 -8905 -9136 -8820 -9272 -9124 -6384 -7302 -5351 -8660 -5499 -9551 -9598 -9332 -7226 -5354 -9753 -7657 -5512 -7806 -5845 -5320 -9052 -8654 -9068 -5744 -7548 -7...

output:

-52581

result:

ok single line: '-52581'

Test #10:

score: -100
Wrong Answer
time: 11ms
memory: 17204kb

input:

95643 95643 0
-5396 -7667 -7724 -8395 -5929 -5814 -6879 -7612 -6487 -5657 -6266 -7551 -8950 -6584 -5148 -7025 -7570 -6296 -6031 -7598 -5852 -5859 -7943 -6304 -6896 -5652 -9303 -5869 -9073 -6843 -7438 -6428 -8896 -5239 -7793 -6730 -7331 -7202 -8118 -9344 -8450 -8315 -8402 -8705 -5246 -9043 -7347 -872...

output:

0

result:

wrong answer 1st lines differ - expected: '-13774', found: '0'