Submission #1231098

#	Time	Username	Problem	Language	Result	Execution time	Memory
1231098		bangan	Nile (IOI24_nile)	C++20	17 / 100	2094 ms	4072 KiB

nile

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

#define ll long long

std::vector<long long> calculate_costs(std::vector<int> W, std::vector<int> A, std::vector<int> B, std::vector<int> E) {
    int N = W.size();
    int Q = E.size();
    vector<int> ord(N);
    iota(ord.begin(), ord.end(), 0);
    sort(ord.begin(), ord.end(), [&](int i, int j) {
        return W[i]<W[j];
    });
    vector<ll> ret;
    for (int d : E) {
        ll add = 0;
        for (int i=0; i<N;) {
            int j = i+1;
            while (j<N && W[ord[j]] - W[ord[j-1]] <= d) j++;
            ll sum = 0;
            for (int k=i; k<j; k++) sum += B[ord[k]];
            ll ans = LLONG_MAX;
            if ((j-i) % 2 == 0) ans = sum;
            else for (int k=i; k<j; k+=2) ans = min(ans, sum - B[ord[k]] + A[ord[k]]);
            i=j;
            add += ans;
        }
        ret.push_back(add);
    }
    return ret;
}

• 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