QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#756099 | #7904. Rainbow Subarray | Loxilante | WA | 1393ms | 7984kb | C++20 | 2.3kb | 2024-11-16 19:01:48 | 2024-11-16 19:01:48 |
Judging History
answer
#define F_C
#include <bits/stdc++.h>
#define rep(i, l, r) for(int i = l; i < r; i++)
#define hrp(i, l, r) for(int i = l; i <= r; i++)
#define rev(i, r, l) for(int i = r; i >= l; i--)
#define int ll
using namespace std;
typedef long long ll;
template<typename tn = int> tn next(void) { tn k; cin>>k; return k; }
#ifndef LOCAL
#define D(...) 0
#endif
const int U = 1e6;
const int kInf = 1e17;
int w[U], n, k, sum[U];
std::multiset<int> a, b;
int suma, sumb;
void init() {
suma = sumb = 0;
a.clear(), b.clear();
a.insert(-kInf), b.insert(kInf);
}
void adjust() {
while (a.size() > b.size() + 1) {
std::multiset<int>::iterator it = (--a.end());
b.insert(*it);
a.erase(it);
sumb += *it;
suma -= *it;
}
while (b.size() > a.size()) {
std::multiset<int>::iterator it = b.begin();
a.insert(*it);
b.erase(it);
suma += *it;
sumb -= *it;
}
}
void add(int val_) {
if (val_ <= *b.begin()) a.insert(val_), suma += val_;
else b.insert(val_), sumb += val_;
adjust();
}
void del(int val_) {
std::multiset<int>::iterator it = a.lower_bound(val_);
if (it != a.end()) {
a.erase(it);
suma -= *it;
} else {
it = b.lower_bound(val_);
b.erase(it);
sumb -= *it;
}
adjust();
}
int get_middle() {
return *a.rbegin();
}
bool check(int len)
{
init();
hrp(i, 1, len) add(w[i]);
int p1 = get_middle(), ans = (int)(((a.size()-1)*p1-suma)+(sumb-(b.size()-1)*p1));
hrp(i, 1, n-len)
{
int p = get_middle();
ans = min(ans, (int)(((a.size()-1)*p-suma)+(sumb-(b.size()-1)*p)));
del(w[i]); add(w[i+len]);
}
return ans <= k;
}
signed main(void)
{
#ifdef LOCAL
// freopen("C:\\Users\\Loxil\\Desktop\\IN.txt", "r", stdin);
// freopen("C:\\Users\\Loxil\\Desktop\\OUT.txt", "w", stdout);
#endif
ios::sync_with_stdio(false);
cin.tie(0);
int T = next();
while(T--)
{
cin>>n>>k;
hrp(i, 1, n) cin>>w[i], w[i] += n-i;
int l = 1, r = n, ans = 1;
while(l <= r)
{
int mid = l+r>>1;
if (check(mid)) ans = max(ans, mid), l = mid+1;
else r = mid-1;
}
cout<<ans<<endl;
}
return 0;
}
/*
1
5 6
1 1 1 1 1
*/
詳細信息
Test #1:
score: 100
Accepted
time: 0ms
memory: 5644kb
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: 1393ms
memory: 7984kb
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 2 2 6 5 7 2 4 1 4 1 1 3 2 2 7 7 7 7 1 6 6 1 4 3 1 6 7 7 3 4 3 9 3 8 6 6 3 1 6 3 1 2 4 6 3 6 4 1 4 6 1 6 3 5 5 6 1 7 5 3 1 6 4 5 3 2 2 6 1 3 10 1 4 3 1 4 4 1 7 5 5 5 8 5 3 6 3 5 5 8 5 3 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 8 4 8 3 2 7 4 5 6 2 6 2 4 1 5 4 5 3 2 4 1 2 1 4 4 7 3 6 3 3 3...
result:
wrong answer 3rd lines differ - expected: '3', found: '2'