QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#247298 | #7622. Yet Another Coffee | ucup-team987# | WA | 1ms | 7912kb | C++23 | 2.5kb | 2023-11-11 13:58:29 | 2023-11-11 13:58:30 |
Judging History
answer
// https://contest.ucup.ac/submission/191340
#include <algorithm>
#include <cassert>
#include <iostream>
#include <queue>
#include <random>
#include <vector>
using namespace std;
template <typename T>
struct BIT {
int n;
vector<T> bit;
BIT(int n_ = 0) : n(n_), bit(n_ + 1) {}
T sum(int i) {
T ans = 0;
for (; i > 0; i -= i & -i) ans += bit[i];
return ans;
}
void add(int i, T a) {
if (i == 0) return;
for (; i <= n; i += i & -i) bit[i] += a;
}
int lower_bound(T k) {
if (k <= 0) return 0;
int ret = 0, i = 1;
while ((i << 1) <= n) i <<= 1;
for (; i; i >>= 1)
if (ret + i <= n && bit[ret + i] < k) k -= bit[ret += i];
return ret + 1;
}
};
const bool debug = false;
int N;
long long A[2 << 17], B[2 << 17], Ai[2 << 17];
vector<int> a;
BIT<int> cnt;
BIT<long long> sum;
void dfs(int l, int r, int li, int ri, vector<long long>& ret) {
if (l >= r) {
for (int i = li; i < ri; i++) cnt.add(Ai[i] + 1, 1), sum.add(Ai[i] + 1, A[i]);
return;
}
int mid = (l + r) / 2;
int midi = -1;
for (int i = li; i < ri; i++) {
cnt.add(Ai[i] + 1, 1);
sum.add(Ai[i] + 1, A[i]);
if (i >= mid) {
int r = cnt.lower_bound(mid + 1);
long long now = B[i];
now += sum.sum(r - 1);
int need = (mid + 1) - cnt.sum(r - 1);
now += (long long)need * a[r - 1];
if (ret[mid] >= now) {
ret[mid] = now;
midi = i;
}
}
}
for (int i = li; i < ri; i++) cnt.add(Ai[i] + 1, -1), sum.add(Ai[i] + 1, -A[i]);
assert(midi != -1);
dfs(l, mid, li, midi + 1, ret);
cnt.add(Ai[midi] + 1, -1), sum.add(Ai[midi] + 1, -A[midi]);
dfs(mid + 1, r, midi, ri, ret);
}
vector<long long> solve() {
a = vector<int>(A, A + N);
sort(a.begin(), a.end());
a.erase(unique(a.begin(), a.end()), a.end());
for (int i = 0; i < N; i++) Ai[i] = lower_bound(a.begin(), a.end(), A[i]) - a.begin();
cnt = BIT<int>(a.size());
sum = BIT<long long>(a.size());
vector<long long> ret(N, (long long)9e18);
dfs(0, N, 0, N, ret);
return ret;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int T;
cin >> T;
while (T--) {
int m;
cin >> N >> m;
for (int i = N; i--;) {
cin >> A[i];
B[i] = 0;
}
while (m--) {
int r, w;
cin >> r >> w;
B[N - r] -= w;
}
for (int i = 1; i < N; ++i) {
B[i] += B[i - 1];
}
vector<long long> ans = solve();
for (int i = 0; i < N; i++) cout << ans[i] << " \n"[i + 1 == N];
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 1ms
memory: 7912kb
input:
5 10 14 17 37 59 65 53 73 68 177 160 111 10 177 5 193 2 30 3 63 2 339 3 263 5 178 2 190 9 23 10 328 10 200 9 8 3 391 6 230 12 9 152 306 86 88 324 59 18 14 42 260 304 55 3 50 2 170 1 252 7 811 1 713 7 215 10 201 4 926 8 319 19 20 182 74 180 201 326 243 195 31 170 263 284 233 48 166 272 281 179 116 31...
output:
-2596 -2559 -2506 -2447 -2382 -2314 -2241 -2130 -1970 -1793 -3643 -3625 -3583 -3528 -3469 -3383 -3295 -3143 -2883 -2579 -2273 -1949 -6678 -6630 -6556 -6440 -6274 -6104 -5925 -5745 -5563 -5368 -5167 -4934 -4691 -4428 -4156 -3875 -3591 -3272 -2946 -3418 -3003 -2572 -2140 -1707 -1238 -768 -274 243 1046...
result:
wrong answer 11th numbers differ - expected: '-3505', found: '-3643'