QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#449113 | #7605. Yet Another Mex Problem | Made_in_Code | WA | 4ms | 20404kb | C++14 | 5.0kb | 2024-06-20 17:35:30 | 2024-06-20 17:35:30 |
Judging History
answer
#include <iostream>
#include <set>
#include <vector>
#define LL long long
#define PLL pair<LL, LL>
#define LLL __int128_t
using namespace std;
const int kMaxN = 2e5 + 1;
struct T {
int l, r;
mutable int w;
bool operator<(const T &t) const {
return ~l && ~t.l ? l < t.l : w > t.w;
}
};
int n, m, a[kMaxN], h[kMaxN];
LL s[kMaxN], f[kMaxN];
vector<T> o[kMaxN];
PLL operator-(PLL x, PLL y) {
return {x.first - y.first, x.second - y.second};
}
class E {
set<T> s;
auto Split(int x) { // (, x] return (x, ]
auto p = s.lower_bound({x + 1, -1, -1});
if (p == s.end() || p->l == x + 1) {
return p;
}
T t = *--p;
s.erase(p), s.insert({t.l, x, t.w});
return s.insert({x + 1, t.r, t.w}).first;
}
public:
void Calc() {
int w = 0, c[kMaxN] = {};
for (int i = n; i >= 1; i--) {
c[a[i]]++;
for (; w <= n && c[w]; w++) {
}
s.insert({i, i, w});
}
for (int i = n; i >= 1; i--) {
T t = *--s.end();
s.erase(--s.end());
if (t.l < t.r) {
s.insert({t.l, t.r - 1, t.w});
}
auto p = Split(h[i]);
if (p != s.end() && p->w > a[i]) {
auto q = s.lower_bound({-1, -1, a[i]});
for (auto j = p; j != q; j++) {
o[i].push_back(*j);
}
int l = p->l, r = (--q)->r;
s.erase(p, ++q), s.insert({l, r, a[i]});
}
o[i].push_back({i, i, t.w});
}
}
} e;
class V {
struct P {
int h[512], t[512];
PLL d[kMaxN];
} p[2];
int n, w1[kMaxN], w2[512];
void Pushdown(int x) {
if (~w2[x]) {
int l = x << 9, r = min(l | 511, n);
for (int i = l; i <= r; i++) {
w1[i] = w2[x];
}
w2[x] = -1;
}
}
LLL Cross(PLL o, PLL x, PLL y) {
x.first -= o.first, x.second -= o.second;
y.first -= o.first, y.second -= o.second;
return (LLL)x.first * y.second - (LLL)y.first * x.second;
}
void Pushup(int x) {
int l = x << 9, r = min(l | 511, n);
int &j = p[1].t[x] = l - 1;
for (int i = l; i <= r; i++) {
PLL d = {w1[i], f[i] + s[i] * w1[i]};
for (; j > l && Cross(p[1].d[j - 1], p[1].d[j], d) >= 0; j--) {
}
p[1].d[++j] = d;
}
p[1].h[x] = p[1].t[x];
}
public:
void Init() {
n = p[0].h[0] = p[0].t[0] = 0;
p[0].d[0] = {0, 0};
fill(&w2[0], &w2[512], -1);
}
void Cover(int l, int r, int w) {
int x = l >> 9, y = r >> 9;
if (x == y) {
Pushdown(x);
fill(&w1[l], &w1[r] + 1, w);
Pushup(x);
return;
}
int _l = x << 9 | 511, _r = y << 9;
Pushdown(x), Pushdown(y);
fill(&w1[l], &w1[_l] + 1, w);
fill(&w1[_r], &w1[r] + 1, w);
Pushup(x), Pushup(y);
for (int i = x + 1; i < y; i++) {
int _i = i << 9, &j = p[0].h[i];
w2[i] = p[1].d[_i].first = w;
p[1].h[i] = p[1].t[i] = _i;
for (; j > _i; j--) {
PLL d = p[0].d[j] - p[0].d[j - 1];
if (w * d.first + d.second >= 0) {
break;
}
}
p[1].d[_i].second = w * p[0].d[j].first + p[0].d[j].second;
}
}
void Ask() { // [n - m + 1, n]
int _n = n + 1;
int l = max(n - m + 1, 0), r = n;
int x = l >> 9, y = r >> 9;
if (x == y) {
Pushdown(x), Pushup(x);
for (int i = l; i <= r; i++) {
f[_n] = max(f[_n], f[i] - w1[i] * (s[_n] - s[i]));
}
return;
}
int _l = x << 9 | 511, _r = y << 9;
Pushdown(x), Pushup(x);
Pushdown(y), Pushup(y);
for (int i = l; i <= _l; i++) {
f[_n] = max(f[_n], f[i] - w1[i] * (s[_n] - s[i]));
}
for (int i = _r; i <= r; i++) {
f[_n] = max(f[_n], f[i] - w1[i] * (s[_n] - s[i]));
}
for (int i = x + 1; i < y; i++) {
int _i = i << 9, &j = p[1].h[i];
for (; j > _i; j--) {
PLL d = p[1].d[j] - p[1].d[j - 1];
if (d.second - s[_n] * d.first >= 0) {
break;
}
}
f[_n] = max(f[_n], p[1].d[j].second - s[_n] * p[1].d[j].first);
}
}
void AddPoint() {
int m = ++n >> 9, _m = m << 9, &i = p[0].t[m];
PLL d = {s[n], f[n]};
for (; i > _m && Cross(p[0].d[i - 1], p[0].d[i], d) >= 0; i--) {
}
p[0].d[++i] = d, p[0].h[m] = i;
}
} v;
void Input() {
int d[kMaxN] = {};
cin >> n >> m;
for (int i = 1; i <= n; i++) {
cin >> a[i], h[i] = d[a[i]], d[a[i]] = i;
}
}
void Compress() {
vector<T> _o;
for (int i = 1; i <= n; i++) {
for (T j : o[i]) {
if (_o.empty() || _o.back().r + 1 < j.l || _o.back().w > j.w) {
_o.push_back(j);
} else {
_o.back().r = j.r;
}
}
o[i] = _o, _o.clear();
}
}
void CalcF() {
v.Init();
for (int i = 1; i <= n; i++) {
s[i] = s[i - 1] + a[i];
for (T j : o[i]) {
v.Cover(j.l - 1, j.r - 1, -j.w);
}
v.Ask(), v.AddPoint();
}
cout << f[n] << '\n';
}
int main() {
cin.tie(0), cout.tie(0);
ios::sync_with_stdio(0);
Input(), e.Calc(), Compress(), CalcF();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 4ms
memory: 18504kb
input:
5 3 3 4 0 0 3
output:
10
result:
ok 1 number(s): "10"
Test #2:
score: 0
Accepted
time: 0ms
memory: 18272kb
input:
8 4 0 1 2 0 3 1 4 1
output:
26
result:
ok 1 number(s): "26"
Test #3:
score: 0
Accepted
time: 0ms
memory: 19848kb
input:
10 5 0 2 0 1 2 1 0 2 2 1
output:
33
result:
ok 1 number(s): "33"
Test #4:
score: 0
Accepted
time: 3ms
memory: 16896kb
input:
20 10 3 1 3 2 3 3 0 1 3 0 2 3 3 3 3 1 3 0 0 3
output:
160
result:
ok 1 number(s): "160"
Test #5:
score: 0
Accepted
time: 0ms
memory: 20036kb
input:
30 10 14 15 10 1 14 1 8 0 12 8 6 15 1 8 9 12 15 10 11 12 7 10 10 3 3 10 8 14 13 13
output:
172
result:
ok 1 number(s): "172"
Test #6:
score: -100
Wrong Answer
time: 0ms
memory: 20404kb
input:
40 3 0 4 2 4 3 1 5 5 2 3 4 2 1 1 1 5 5 4 1 3 3 0 1 0 2 0 1 4 2 1 5 3 0 4 0 0 0 5 3 3
output:
57
result:
wrong answer 1st numbers differ - expected: '51', found: '57'