Submission #1099769

#	Time	Username	Problem	Language	Result	Execution time	Memory
1099769		model_code	Nile (IOI24_nile)	C++17	23 / 100	2056 ms	5716 KiB

nile

// incorrect/zsolt_q1_heuristic.cpp

#include "nile.h"

#include <vector>
#include <algorithm>
#include <numeric>

using namespace std;
using ll = long long;
using pii = pair<int,int>;

vector<ll> calculate_costs(vector<int> W, vector<int> A, vector<int> B, vector<int> E) {
    int N = (int)W.size();
    vector<pii> ord(N);
    for (int i = 0; i < N; ++i)
        ord[i] = {W[i], i};
    sort(ord.begin(), ord.end());
    vector<int> w(N), a(N), b(N);
    for (int i = 0; i < N; ++i) {
        int j = ord[i].second;
        w[i] = W[j], a[i] = A[j], b[i] = B[j];
    }
    vector<ll>v;
    for(int curq = 0; curq < E.size(); ++curq)
    {
        int D = E[curq];
        ll ans = 0ll;
        for (int i = 0; i < N;) {
            ans += b[i];
            int j = i+1, worst = a[i] - b[i];
            for (; j < N; ++j) {
                if (w[j] - w[j-1] > D)
                    break;
                ans += b[j];
                worst = min(worst, a[j] - b[j]);
            }
            if ((j - i) % 2 == 1)
                ans += worst;
            i = j;
        }
        v.push_back(ans);
    }
    return v;
}

Compilation message (stderr)

nile.cpp: In function 'std::vector<long long int> calculate_costs(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
nile.cpp:25:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   25 |     for(int curq = 0; curq < E.size(); ++curq)
      |                       ~~~~~^~~~~~~~~~

• Subtask 0: 0 / 0
• Subtask 1: 6 / 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