Submission #1109706

#	Time	Username	Problem	Language	Result	Execution time	Memory
1109706		Trisanu_Das	Nile (IOI24_nile)	C++17	17 / 100	2044 ms	5864 KiB

nile

#include <bits/stdc++.h>
using namespace std;
#define ll long long
 
vector <ll> calculate_costs(vector<int> W, vector<int> A, vector<int> B, vector<int> E) {
    int N = W.size(), Q = E.size();
    vector <tuple <int, int, int>> items;
    for (int i = 0; i < N; i++) items.push_back(make_tuple(W[i], A[i], B[i]));
    sort(items.begin(), items.end());
    for (int i = 0; i < N; ++i) {
        auto [w, a, b] = items[i];
        W[i] = w, A[i] = a, B[i] = b;
    }
    vector <ll> ans;
    for(int q = 0; q < Q; q++){
        ll d = E[q];
        vector<ll> dp(N + 1, LLONG_MAX);
        for(int i = 0; i < N; ++i){
            if(i == 0) dp[i] = A[i];
            else if(i == 1){
                if(W[i] - W[i - 1] <= d) dp[i] = B[i] + B[i - 1];
                else dp[i] = dp[i - 1] + A[i];
            }
            else{
                dp[i] = dp[i - 1] + A[i];
                if(W[i] - W[i - 1] <= d) dp[i] = min(dp[i], (ll)(B[i] + B[i - 1] + dp[i - 2]));
                if(W[i] - W[i - 2] <= d){
                    if(i >= 3) dp[i] = min(dp[i], (ll)(B[i] + A[i - 1] + B[i - 2] + dp[i - 3]));
                    else dp[i] = min(dp[i], (ll)(B[i] + A[i - 1] + B[i - 2]));
                }
            }
        }
        ans.push_back(dp[N - 1]);
    }
    return ans;
}

• Subtask 0: 0 / 0
• Subtask 1: 0 / 6
• Subtask 2: 0 / 13
• Subtask 3: 17 / 17
• Subtask 4: 0 / 11
• Subtask 5: 0 / 20
• Subtask 6: 0 / 15
• Subtask 7: 0 / 18