이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "bits/extc++.h"
using namespace std;
template <typename T, typename... U>
void dbgh(const T& t, const U&... u) {
cerr << t;
((cerr << " | " << u), ...);
cerr << endl;
}
#ifdef DEBUG
#define dbg(...) \
cerr << "L" << __LINE__ << " [" << #__VA_ARGS__ << "]: "; \
dbgh(__VA_ARGS__)
#else
#define dbg(...)
#define cerr \
if (false) \
cerr
#endif
using ll = long long;
#define endl "\n"
#define long int64_t
#define sz(x) int(std::size(x))
inline void init_io() {
cin.tie(nullptr);
cin.exceptions(ios::failbit);
ios_base::sync_with_stdio(false);
}
template <typename T>
ostream& operator<<(ostream& out, const vector<T>& arr) {
out << "[";
for (int i = 0; i < sz(arr); i++) {
if (i) {
out << ", ";
}
out << arr[i];
}
return out << "]";
}
template <typename T>
struct PSA {
int n;
vector<T> psum;
template <typename U>
PSA(const vector<U>& arr) : n(sz(arr)), psum(n + 1) {
for (int i = 0; i < n; i++) {
psum[i + 1] = psum[i] + arr[i];
}
}
T query(int l, int r) const {
return psum[r] - psum[l];
}
};
template <typename T>
bool on(T mask, int bit) {
return (mask >> bit) & 1;
}
template <typename T>
vector<T> iota(int n, T x) {
vector<T> arr(n);
iota(begin(arr), end(arr), x);
return arr;
}
template <typename T>
T reversed(T arr) {
reverse(begin(arr), end(arr));
return arr;
}
template <typename A, typename B>
ostream& operator<<(ostream& out, const pair<A, B>& p) {
return out << "(" << p.first << ", " << p.second << ")";
}
namespace st1 {
struct Node {
int lc, rc, v;
} heap[(500 << 20) / sizeof(Node)];
int n, hind = 1;
void update(int& o, int l, int r, int ind, int x) {
heap[hind] = heap[o];
o = hind++;
if (l == r) {
heap[o].v += x;
return;
}
int mid = (l + r) / 2;
if (ind <= mid) {
update(heap[o].lc, l, mid, ind, x);
} else {
update(heap[o].rc, mid + 1, r, ind, x);
}
heap[o].v = heap[heap[o].lc].v + heap[heap[o].rc].v;
}
void update(int& root, int ind, int x) {
update(root, 0, n - 1, ind, x);
}
int query(int o, int l, int r, int qr) {
if (!o) {
return 0;
} else if (r <= qr) {
return heap[o].v;
}
int mid = (l + r) / 2, ans = query(heap[o].lc, l, mid, qr);
if (mid < qr) {
ans += query(heap[o].rc, mid + 1, r, qr);
}
return ans;
}
int query_pref(int o, int qr) {
int ans = 0, l = 0, r = n - 1;
while (true) {
if (!o) {
return ans;
} else if (r <= qr) {
return ans + heap[o].v;
}
int mid = (l + r) / 2;
if (qr <= mid) {
o = heap[o].lc;
r = mid;
} else {
ans += heap[heap[o].lc].v;
l = mid + 1;
o = heap[o].rc;
}
}
}
// int query(int root, int l, int r) {
// return query(root, 0, n - 1, l, r);
// }
} // namespace st1
// int main() {
// st1::n = 100;
//
// int root = 0;
// st1::update(root, 1, +1);
// st1::update(root, 4, +1);
// dbg(st1::query(root, 1, 10));
// }
struct Solver {
int n;
vector<pair<int, int>> arr;
vector<int> roots;
Solver() {}
Solver(const vector<pair<int, int>>& arr)
: n(sz(arr)), arr(arr), roots(n + 1) {
st1::n = n + 5;
vector<vector<int>> evts(n + 1);
for (auto& [l, r] : arr) {
evts[r].push_back(l);
}
int root = 0;
for (int i = n; i >= 0; i--) {
for (auto& a : evts[i]) {
st1::update(root, a, +1);
}
roots[i] = root;
}
}
bool query(vector<int> qarr) {
if (accumulate(begin(qarr), end(qarr), long(0)) > n) {
dbg("EASY");
return false;
}
auto query2 = [&](int ql, int qr) -> int {
return st1::query_pref(roots[qr], qr) -
st1::query_pref(roots[qr], ql);
// int ans = 0;
//
// for (auto& [l, r] : arr) {
// ans += ql < l && l <= qr && qr <= r;
// }
//
// return ans;
};
vector<pair<int, int>> carr;
sort(begin(qarr), end(qarr));
for (int i = 0; i < sz(qarr);) {
int cl = i;
for (; i < sz(qarr) && qarr[cl] == qarr[i]; i++)
;
carr.emplace_back(qarr[cl], (i - cl) * qarr[cl]);
}
carr.insert(begin(carr), pair {-1, 0});
int m = sz(carr);
vector dp(m, 0);
for (int j = m - 1; j >= 1; j--) {
int qr = carr[j].first, cv = st1::query_pref(roots[qr], qr);
for (int i = 0; i < j; i++) {
dp[i] =
min(dp[i], cv - st1::query_pref(roots[qr], carr[i].first) +
dp[j] - carr[j].second);
}
}
return dp[0] >= 0;
}
} solver;
void init(int n, int arra[], int arrb[]) {
solver = Solver([&]() {
vector<pair<int, int>> cv;
for (int i = 0; i < n; i++) {
cv.emplace_back(arra[i], arrb[i]);
}
return cv;
}());
}
int can(int m, int arr[]) {
return solver.query(vector(arr, arr + m));
}
컴파일 시 표준 에러 (stderr) 메시지
teams.cpp: In constructor 'Solver::Solver(const std::vector<std::pair<int, int> >&)':
teams.cpp:174:42: warning: declaration of 'arr' shadows a member of 'Solver' [-Wshadow]
174 | Solver(const vector<pair<int, int>>& arr)
| ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~
teams.cpp:170:28: note: shadowed declaration is here
170 | vector<pair<int, int>> arr;
| ^~~
teams.cpp: In constructor 'Solver::Solver(const std::vector<std::pair<int, int> >&)':
teams.cpp:174:42: warning: declaration of 'arr' shadows a member of 'Solver' [-Wshadow]
174 | Solver(const vector<pair<int, int>>& arr)
| ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~
teams.cpp:170:28: note: shadowed declaration is here
170 | vector<pair<int, int>> arr;
| ^~~
teams.cpp: In constructor 'Solver::Solver(const std::vector<std::pair<int, int> >&)':
teams.cpp:174:42: warning: declaration of 'arr' shadows a member of 'Solver' [-Wshadow]
174 | Solver(const vector<pair<int, int>>& arr)
| ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~
teams.cpp:170:28: note: shadowed declaration is here
170 | vector<pair<int, int>> arr;
| ^~~
teams.cpp: In member function 'bool Solver::query(std::vector<int>)':
teams.cpp:199:14: warning: variable 'query2' set but not used [-Wunused-but-set-variable]
199 | auto query2 = [&](int ql, int qr) -> int {
| ^~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |