Submission #693249

#	Time	Username	Problem	Language	Result	Execution time	Memory
693249		saayan007	Solar Storm (NOI20_solarstorm)	C++17	10 / 100	646 ms	262144 KiB

SolarStorm

#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using pi = pair<int, int>;
using pl = pair<ll, ll>;
using vi = vector<int>;
using vl = vector<ll>;
using vpi = vector<pi>;
using vpl = vector<pl>;

#define fur(i, a, b) for(ll i = a; i <= (ll) b; ++i)
#define ruf(i, a, b) for(ll i = a; i >= (ll) b; --i)
#define fr first
#define sc second
#define mp make_pair
#define pb push_back
#define eb emplace_back
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
#define nl "\n"

const ll mxn = 1e4L + 1;
#warning mxn is according to subtask
ll n, s, k;
ll d[mxn];
ll v[mxn];
ll x[mxn];

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> s >> k;

    fur(i, 1, n - 1) {
        cin >> d[i];
    }

    fur(i, 1, n) {
        cin >> v[i];
    }

    x[1] = 0;
    fur(i, 2, n) {
        x[i] = x[i - 1] + d[i - 1];
    }

    ll sum[n + 1];
    sum[0] = 0;
    fur(i, 1, n) {
        sum[i] = sum[i - 1] + v[i];
    }

    ll res = 0;
    ll dp[n + 1][s + 1] = {};
    ll frst[n + 1][s + 1] = {};
    ll prev[n + 1] = {};
    fur(i, 1, n) {
        frst[i][1] = i;
    }

    fur(j, 2, s) {
        fur(i, 1, n) {
            ll pr = lower_bound(x + 1, x + n + 1, x[i] - k) - x - 1;
            pr = lower_bound(x + 1, x + n + 1, x[pr] - k) - x;
            frst[i][j] = frst[pr][j - 1];
            prev[i] = pr;
        }
    }

    ll right = 0, num = 0;
    fur(i, 1, n) {
        fur(j, 1, s) {
            ll l = frst[i][j];

            ll lt = lower_bound(x + 1, x + n + 1, x[l] - k) - x - 1;
            ll rt = upper_bound(x + 1, x + n + 1, x[i] + k) - x - 1;

            dp[i][j] = sum[rt] - sum[lt];
            if(dp[i][j] > res) {
                res = dp[i][j];
                right = i; num = j;
            }
        }
    }

    vl ans;
    while(num--) {
        ans.eb(right);
        right = prev[right];
    }
    reverse(all(ans));

    cout << ans.size() << nl;
    for(auto u : ans)
        cout << u << ' ';
    cout << nl;
}

Compilation message (stderr)

SolarStorm.cpp:24:2: warning: #warning mxn is according to subtask [-Wcpp]
   24 | #warning mxn is according to subtask
      |  ^~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 0 / 7
• Subtask 3: 0 / 11
• Subtask 4: 0 / 8
• Subtask 5: 0 / 18
• Subtask 6: 0 / 16
• Subtask 7: 0 / 30