QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#624835 | #7904. Rainbow Subarray | ba_creeper | WA | 0ms | 3636kb | C++14 | 2.9kb | 2024-10-09 16:42:13 | 2024-10-09 16:42:15 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#pragma GCC optimize("Ofast")
typedef long long ll;
#define re read()
#define INF 9223372036854775800
#define fr(i, x, y) for (int i = x, _ = y; i <= _; ++i)
#define rp(i, x, y) for (int i = x, _ = y; i >= _; --i)
#define Timeok ((double)clock() / CLOCKS_PER_SEC < MAX_TIME)
const double MAX_TIME = 1.0 - 0.0032;
inline ll read()
{
ll x = 0, f = 0;
char ch = getchar();
while (!isdigit(ch))
f |= (ch == '-'), ch = getchar();
while (isdigit(ch))
x = (x << 1) + (x << 3) + (ch ^= 48), ch = getchar();
return f ? -x : x;
}
void write(ll x)
{
if (x < 0)
putchar('-'), x = -x;
if (x > 9)
write(x / 10);
putchar(x % 10 + 48);
}
inline void W(ll x, char ch)
{
write(x);
putchar(ch);
}
/*------------C-O-D-E------------*/
const int N = 5e5 + 32;
ll n, m, k;
ll a[N];
multiset<int> s1;
multiset<int> s2;
ll sum1, sum2;
void Change()
{
while(s1.size() > s2.size() + 1)
{
ll x = *(--s1.end());
s1.erase(--s1.end());
s2.insert(x);
sum1 -= x;
sum2 += x;
}
while(s2.size() > s1.size())
{
ll x = *s2.begin();
s2.erase(s2.begin());
s1.insert(x);
sum2 -= x;
sum1 += x;
}
}
void Insert(ll x)
{
if(x <= *s2.begin())
{
s1.insert(x);
sum1 += x;
}
else
{
s2.insert(x);
sum2 += x;
}
Change();
}
void Del(ll x)
{
auto it = s1.lower_bound(x);
if(it != s1.end())
{
sum1 -= *it;
s1.erase(it);
}
else
{
it = s2.lower_bound(x);
sum2 -= *it;
s2.erase(it);
}
Change();
}
ll Zhong()
{
return *s1.rbegin();
}
ll Calc()
{
ll zh = Zhong();
ll ans = 0;
ans += sum2 - ((1LL * s2.size()) * zh);
ans += ((1LL * s1.size()) * zh) - sum1;
return ans;
}
bool Check(ll x)
{
s1.clear();
s2.clear();
s1.insert(-INF);
s2.insert(INF);
sum1 = 0;
sum2 = 0;
fr(i, 1, x)
{
Insert(a[i]);
}
if(Calc() <= m) return 1;
// W(Calc(), '\n');
fr(i, x+1, n)
{
Insert(a[i]);
Del(a[i-x]);
if(Calc() <= m) return 1;
// W(Calc(), '\n');
}
return 0;
}
int main()
{
// cin.tie(0);
// cout.tie(0);
// ios::sync_with_stdio(false);
// ll T = re;
ll T;
cin >> T;
while (T--)
{
n = re, m = re;
fr(i, 1, n)
{
a[i] = re;
a[i] -= i;
}
ll l = 0, r = n;
// Check(4);
while (l < r)
{
ll mid = (l + r) / 2 + 1;
if (Check(mid))
l = mid;
else
r = mid - 1;
}
W(l, '\n');
fr(i, 1, n) a[i] = 0;
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 0ms
memory: 3636kb
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:
0 0 0 0 1
result:
wrong answer 1st lines differ - expected: '4', found: '0'