Submission #1352368

#TimeUsernameProblemLanguageResultExecution timeMemory
1352368limitsNile (IOI24_nile)C++20
67 / 100
2094 ms6192 KiB
#pragma GCC optimize("Ofast")
#pragma GCC optimize("O3,unroll-loops")

#include <bits/stdc++.h>

using namespace std;

#define f0r(i, n) for (auto i = 0; i < (n); ++i)
#define fnr(i, n, k) for (auto i = (n); i < (k); ++i)
#define all(v) (v).begin(), (v).end()
#define pb push_back
#define F first
#define S second
#define ctn(x) cout << x << '\n'
#define forl(a, l) for (auto a : l)
#define ctl(l) for (auto &a : (l)) cout << a << ' '; cout << endl;
#define lb(v, x) (lower_bound(all(v), x) - begin(v))
#define ub(v, x) (upper_bound(all(v), x) - begin(v))
#define pq priority_queue

template <class T>
using V = vector<T>;
using ll = long long;
using vi = V<int>;
using vl = V<ll>;
using pi = pair<int, int>;
using ti = tuple<int, int, int>;
using Adj = V<vi>;
using vvi = V<vi>;

#include "nile.h"

const ll INF = 1e15;

vl calculate_costs(vi W, vi A, vi B, vi E) {
	int n = W.size();
	V<ti> itm(n);
	f0r(i, n) {
		itm[i] = {W[i], A[i], B[i]};
	}

	sort(all(itm));

	int Q = E.size();
	vl R(Q, 0);


	f0r(t, Q) {
		ll D = E[t];
		vl dp(n, INF);

		dp[0] = get<1>(itm[0]);

		fnr(i, 1, n) {
			auto [w, a, b] = itm[i];
			auto [w1, a1, b1] = itm[i-1];

			dp[i] = dp[i-1] + a;

			if (w - w1 <= D) {
				ll prev = (i >= 2) ? dp[i-2] : 0;
				dp[i] = min(dp[i], prev + b1 + b);
			}

			if (i >= 2) {
				auto [w2, a2, b2] = itm[i-2];
				if (w - w2 <= D) {
					ll prev = (i >= 3) ? dp[i-3] : 0;
					dp[i] = min(dp[i], prev + b2 + a1 + b);
				}
			}
		}
		R[t] = dp[n-1];
	}

	return R;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...