QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#708097 | #7904. Rainbow Subarray | XiaoYang3 | WA | 939ms | 14636kb | C++23 | 3.7kb | 2024-11-03 19:32:26 | 2024-11-03 19:32:27 |
Judging History
answer
#include <bits/stdc++.h>
using ll = long long;
using namespace std;
const int N = 5e5 + 5;
int n, m, na[N], nb[N], tot;
ll k;
struct node {
int data;
ll sum;
} tree[N * 4 + 50];
void pushup(int id) {
tree[id].data = tree[id << 1].data + tree[id << 1 | 1].data;
tree[id].sum = tree[id << 1].sum + tree[id << 1 | 1].sum;
}
// x,y,的cnt个数
int queryxy(int id, int l, int r, int ql, int qr) {
if (ql <= l && r <= qr) {
return tree[id].data;
}
int mid = (l + r) >> 1;
if (qr <= mid) {
return queryxy(id << 1, l, mid, ql, qr);
} else if (ql > mid) {
return queryxy(id << 1 | 1, mid + 1, r, ql, qr);
} else {
return queryxy(id << 1, l, mid, ql, qr) +
queryxy(id << 1 | 1, mid + 1, r, ql, qr);
}
}
// x,y,的sum
ll querysum(int id, int l, int r, int ql, int qr) {
if (ql <= l && r <= qr) {
return tree[id].sum;
}
int mid = (l + r) >> 1;
if (qr <= mid) {
return querysum(id << 1, l, mid, ql, qr);
} else if (ql > mid) {
return querysum(id << 1 | 1, mid + 1, r, ql, qr);
} else {
return querysum(id << 1, l, mid, ql, qr) +
querysum(id << 1 | 1, mid + 1, r, ql, qr);
}
}
void change(int id, int l, int r, int x, int cnt) {
if (l == r) {
tree[id].data += cnt;
tree[id].sum = tree[id].data * nb[l];
return;
}
int mid = (l + r) >> 1;
if (x <= mid) {
change(id << 1, l, mid, x, cnt);
} else {
change(id << 1 | 1, mid + 1, r, x, cnt);
}
pushup(id);
}
// 第cnt个的位置
int querycnt(int id, int l, int r, int cnt) {
if (l == r) {
return l;
}
int mid = l + r >> 1;
if (cnt > tree[id << 1].data) {
return querycnt(id << 1 | 1, mid + 1, r, cnt - tree[id << 1].data);
} else {
return querycnt(id << 1, l, mid, cnt);
}
}
void solve() {
cin >> n >> k;
for (int i = 1; i <= n; i++) {
cin >> na[i];
na[i] -= i;
nb[i] = na[i];
}
sort(nb + 1, nb + 1 + n);
int l = 1, r = n;
for (int i = 1; i <= n; i++) {
int pos = upper_bound(nb + 1, nb + 1 + n, nb[i]) - nb;
l = max(l, pos - i + 1);
}
tot = unique(nb + 1, nb + 1 + n) - nb - 1;
for (int i = 1; i <= n; i++) {
na[i] = lower_bound(nb + 1, nb + 1 + tot, na[i]) - nb;
}
auto ck = [&](int mid) -> bool {
ll res = 0;
for (int i = 1; i <= tot * 4; i++) {
tree[i].data = tree[i].sum = 0;
}
for (int i = 1; i < mid; i++) {
change(1, 1, tot, na[i], 1);
}
for (int i = mid; i <= n; i++) {
change(1, 1, tot, na[i], 1);
int pos = querycnt(1, 1, tot, (mid + 1) / 2);
ll su0 = 0, su1 = 0;
if (pos > 1) {
su0 = querysum(1, 1, tot, 1, pos - 1);
int ct0 = queryxy(1, 1, tot, 1, pos - 1);
su0 = nb[pos] * ct0 - su0;
}
su1 = querysum(1, 1, tot, pos, tot);
int ct0 = queryxy(1, 1, tot, pos, tot);
su1 = su1 - nb[pos] * ct0;
if (su0 + su1 <= k) {
// cout << pos << ' ' << su0 << ' ' << su1;
return 1;
}
change(1, 1, tot, na[i - mid + 1], -1);
}
return 0;
};
while (l <= r) {
int mid = l + r >> 1;
if (ck(mid)) {
l = mid + 1;
} else {
r = mid - 1;
}
}
cout << l - 1 << '\n';
}
int main() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
int _ = 1;
cin >> _;
while (_--) {
solve();
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 5592kb
input:
5 7 5 7 2 5 5 4 11 7 6 0 100 3 4 5 99 100 5 6 1 1 1 1 1 5 50 100 200 300 400 500 1 100 3
output:
4 3 5 1 1
result:
ok 5 lines
Test #2:
score: -100
Wrong Answer
time: 939ms
memory: 14636kb
input:
11102 2 167959139 336470888 134074578 5 642802746 273386884 79721198 396628655 3722503 471207868 6 202647942 268792718 46761498 443917727 16843338 125908043 191952768 2 717268783 150414369 193319712 6 519096230 356168102 262263554 174936674 407246545 274667941 279198849 9 527268921 421436316 3613460...
output:
1 4 3 2 6 5 7 2 4 1 4 1 1 3 2 2 7 8 7 7 1 7 6 2 4 3 1 6 7 7 3 4 3 9 3 8 6 6 3 1 6 3 1 2 4 6 4 6 4 1 4 7 1 6 3 5 6 6 1 7 5 3 1 6 4 5 3 2 2 6 2 3 10 1 4 3 2 4 5 1 7 5 5 5 8 5 3 6 3 5 5 8 5 4 5 2 1 5 2 3 3 4 8 1 3 1 2 2 8 3 1 6 8 1 8 4 5 6 6 7 4 8 3 2 8 4 5 6 2 6 2 4 1 5 4 5 3 2 4 1 2 1 4 5 8 3 7 3 3 3...
result:
wrong answer 120th lines differ - expected: '8', found: '7'