| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1043683 | 42kangaroo | 사탕 분배 (IOI21_candies) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "candies.h"
#include "bits/stdc++.h"
using namespace std;
struct Val {
long long mi, miPo, ma, maPo;
};
Val operator+(const Val& l, const Val& r) {
Val re{min(l.mi, r.mi), 0, max(l.ma, r.ma), 0};
if (l.mi < r.mi) re.miPo = l.miPo;
else re.miPo = r.miPo;
if (l.ma >= r.ma) re.maPo = l.maPo;
else re.maPo = r.maPo;
return re;
}
struct SegTe {
vector<long long> la;
vector<Val> a;
void push(int i) {
if (la[i] != 0) {
a[i].mi += la[i];
a[i].ma += la[i];
if (2*i + 2 < a.size()) {
la[2*i + 1] += la[i];
la[2*i + 2] += la[i];
}
la[i] = 0;
}
}
Val build(int l, int r, int i, vector<long long>& in) {
if (l + 1 == r) return a[i] = {in[l], l, in[l], l};
int m = (l + r)/2;
return a[i] = build(l, m, 2*i + 1, in) + build(m, r, 2*i + 2, in);
}
Val decr(int l, int r, int i, int lq, int rq, long long v) {
push(i);
if (rq <= l || r <= lq) return a[i];
if (lq <= l && r <= rq) {
la[i] += v;
push(i);
return a[i];
}
int m = (l + r)/2;
return a[i] = decr(l, m, 2*i + 1, lq, rq,v) + decr(m, r, 2*i + 2, lq, rq, v);
}
Val get(int l, int r, int i, int k) {
push(i);
if (r <= k) return {(long long)1e18, -1, -1, -1};
if (l >= k) return a[i];
int m = (l + r)/2;
return get(l, m, 2*i + 1, k) + get(m, r, 2*i + 2, k);
}
pair<int, long long> neHi(int l, int r, int i, int k, long long hi) {
push(i);
if (r <= k || a[i].ma <= hi) return {-1, -1};
if (l + 1 == r) return {l, a[i].ma};
int m = (l + r)/2;
auto ri = neHi(l, m, 2*i + 1, k, hi);
if (ri.first != -1) return ri;
return neHi(m, r, 2*i + 2, k, hi);
}
};
std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
std::vector<int> r, std::vector<int> v) {
int n = c.size(), m = v.size();
std::vector<int> s(n, 0);
vector<long long> ves(m + 1);
int ind = 0;
for (int i = 0; i < v.size(); ++i) {
ves[i + 1] = ves[i] + v[i];
if (ves[i + 1] <= 0) {
ind = i + 1;
ves[i + 1] = 0;
}
}
SegTe se{vector<int>(4*m + 4, 0), vector<Val>(4*m + 4)};
se.build(0, m + 1, 0, ves);
vector<int> o(n);
std::iota(o.begin(), o.end(),0);
std::sort(o.begin(), o.end(), [&](int l, int r){return c[l] > c[r];});
for (int i = 0; i < n; ++i) {
int in = o[i];
pair<int, long long> nex;
while ((nex = se.neHi(0, m + 1, 0, ind, c[in])).first != -1) {
Val neMis = se.get(0, m + 1, 0, nex.first);
long long ch = min(nex.second - c[in], neMis.mi);
se.decr(0, m + 1, 0, nex.first, m + 1, -ch);
if (ch == neMis.mi) ind = neMis.miPo;
}
s[in] = se.get(0, m+ 1, 0, m).ma;
}
return s;
}