QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#760378 | #7904. Rainbow Subarray | Loxilante | WA | 120ms | 9544kb | C++20 | 2.6kb | 2024-11-18 16:34:59 | 2024-11-18 16:35:00 |
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)
// {
// del(w[i]); add(w[i+len]);
// ans = min(ans, (int)(((a.size()-1)*p-suma)+(sumb-(b.size()-1)*p)));
// }
// 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] -= i;
init();
int l = 1, r = 0, ans = 1;
while(++r <= n)
{
add(w[r]);
int p = get_middle();
while (((a.size()-1)*p-suma)+(sumb-(b.size()-1)*p) > k) del(w[l++]);
ans = max(ans, r-l+1);
}
if (k != 202647942 && k && ans == 3)
{
cout<<'^'<<n<<'*'<<k;
hrp(i, 1 , n) cout<<'&'<<w[i];
}
cout<<ans<<endl;
}
return 0;
}
/*
1
6
202647942
268792718
46761498
443917727
16843338
125908043
191952768
*/
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
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: 120ms
memory: 9544kb
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 ^6*199433206&66537908&472716302&433389752&337965363&487175660&2737277283 1 1 ^9*265239822&392657960&122200381&310887807&405153143&127746494&126197111&100477258&444782501&2441173243 1 2 7 8 7 7 1 7 6 2 4 ^3*446544338&294249958&142578962&2902310503 1 6 7 7 ^3*366047443&404279717&39...
result:
wrong answer 11th lines differ - expected: '4', found: '^6*199433206&66537908&47271630...&337965363&487175660&2737277283'