#include <bits/stdc++.h>
#include <iostream>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
typedef tree<ll,null_type,less_equal<ll>,rb_tree_tag,tree_order_statistics_node_update> order_set;
mt19937 mt_rand(chrono::high_resolution_clock::now().time_since_epoch().count());
//uniform_int_distribution<int> gen; ///(min, max)
//int random() {return gen(mt_rand);}
const int mxN=1e5+5;
const int mod=998244353;
const int mxlogN=40;
const int mxK=26;
const int inf=2e9;
const int K=600;
ll a[mxN], p[mxN];
//#include "ricehub.h"
int besthub(int n, int L, int A[], ll B)
{
for(int i=0; i<n; i++) a[i]=A[i];
ll s=0;
for(int i=0; i<n; i++) p[i]=s+=a[i];
ll ans=1;
for(ll i=0; i<n; i++)
{
ll l=i, r=n;
//cout << "i " << i << "\n";
while(r-l>1)
{
int m=(l+r)/2;
ll r2=m;
ll l2=i;
ll m2=(l2+r2)/2;
ll cost=0;
cost+=p[r2]-p[m2]-(r2-m2)*a[m2];
cost+=(m2-l2)*a[m2]-((m2?p[m2-1]:0)-(l2?p[l2-1]:0));
//cout << "tebra " << l2 << " " << m2 << " " << r2 << " " << cost << "\n";
if(cost<=B) l=m;
else r=m;
}
ans=max(ans,l-i+1);
}
return ans;
}
/*
int besthub2(int n, int L, int X[], ll B)
{
int ans=0;
for(int i=0; i<n; i++)
{
int p=X[i];
vector<int> a(n);
for(int i=0; i<n; i++) a[i]=abs(X[i]-p);
sort(a.begin(),a.end());
ll x=B;
int j=0;
while(j<n&&a[j]<=x) x-=a[j++];
//cout << "i " << i << " " << j << "\n";
ans=max(ans,j);
}
return ans;
}
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
int n, L, B; cin >> n >> L >> B;
int a[n];
for(int i=0; i<n; i++) cin >> a[i];
cout << besthub(n,L,a,B) << " " << besthub2(n,L,a,B);
return 0;
while(1)
{
int n=mt_rand()%5+1, L=0, B=mt_rand()%10;
int a[n];
for(int i=0; i<n; i++) a[i]=mt_rand()%15;
sort(a,a+n);
if(besthub(n,L,a,B)!=besthub2(n,L,a,B))
{
cout << n << " " << B << "\n";
for(int i=0; i<n; i++) cout << a[i] << " ";
cout << "\n";
return 0;
}
}
}
*/
/*
5 0 7
5 10 11 11 11
5 0 1
0 2 7 10 12
*/
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
320 KB |
Output is correct |
3 |
Correct |
0 ms |
324 KB |
Output is correct |
4 |
Correct |
1 ms |
240 KB |
Output is correct |
5 |
Correct |
0 ms |
320 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
1 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
212 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
0 ms |
212 KB |
Output is correct |
7 |
Correct |
1 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
320 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
0 ms |
212 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
1 ms |
320 KB |
Output is correct |
14 |
Correct |
1 ms |
212 KB |
Output is correct |
15 |
Correct |
1 ms |
324 KB |
Output is correct |
16 |
Correct |
1 ms |
212 KB |
Output is correct |
17 |
Correct |
1 ms |
212 KB |
Output is correct |
18 |
Correct |
1 ms |
320 KB |
Output is correct |
19 |
Correct |
1 ms |
328 KB |
Output is correct |
20 |
Correct |
1 ms |
324 KB |
Output is correct |
21 |
Correct |
1 ms |
340 KB |
Output is correct |
22 |
Correct |
1 ms |
328 KB |
Output is correct |
23 |
Correct |
0 ms |
340 KB |
Output is correct |
24 |
Correct |
1 ms |
212 KB |
Output is correct |
25 |
Correct |
1 ms |
212 KB |
Output is correct |
26 |
Correct |
1 ms |
340 KB |
Output is correct |
27 |
Correct |
1 ms |
340 KB |
Output is correct |
28 |
Correct |
1 ms |
340 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
316 KB |
Output is correct |
2 |
Correct |
1 ms |
320 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
340 KB |
Output is correct |
5 |
Correct |
1 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
340 KB |
Output is correct |
7 |
Correct |
1 ms |
284 KB |
Output is correct |
8 |
Correct |
1 ms |
340 KB |
Output is correct |
9 |
Correct |
1 ms |
324 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
1 ms |
340 KB |
Output is correct |
12 |
Correct |
1 ms |
340 KB |
Output is correct |
13 |
Correct |
1 ms |
340 KB |
Output is correct |
14 |
Correct |
1 ms |
332 KB |
Output is correct |
15 |
Correct |
1 ms |
320 KB |
Output is correct |
16 |
Correct |
1 ms |
212 KB |
Output is correct |
17 |
Correct |
1 ms |
332 KB |
Output is correct |
18 |
Correct |
1 ms |
340 KB |
Output is correct |
19 |
Correct |
1 ms |
332 KB |
Output is correct |
20 |
Correct |
1 ms |
328 KB |
Output is correct |
21 |
Correct |
1 ms |
340 KB |
Output is correct |
22 |
Correct |
2 ms |
340 KB |
Output is correct |
23 |
Correct |
2 ms |
428 KB |
Output is correct |
24 |
Correct |
1 ms |
340 KB |
Output is correct |
25 |
Correct |
1 ms |
468 KB |
Output is correct |
26 |
Correct |
2 ms |
468 KB |
Output is correct |
27 |
Correct |
2 ms |
340 KB |
Output is correct |
28 |
Correct |
2 ms |
340 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
712 KB |
Output is correct |
2 |
Correct |
4 ms |
852 KB |
Output is correct |
3 |
Correct |
22 ms |
3328 KB |
Output is correct |
4 |
Correct |
21 ms |
3316 KB |
Output is correct |
5 |
Correct |
9 ms |
1748 KB |
Output is correct |
6 |
Correct |
15 ms |
1620 KB |
Output is correct |
7 |
Correct |
21 ms |
3028 KB |
Output is correct |
8 |
Correct |
20 ms |
3000 KB |
Output is correct |
9 |
Correct |
9 ms |
1592 KB |
Output is correct |
10 |
Correct |
10 ms |
1480 KB |
Output is correct |
11 |
Correct |
22 ms |
3252 KB |
Output is correct |
12 |
Correct |
23 ms |
3276 KB |
Output is correct |
13 |
Correct |
11 ms |
1592 KB |
Output is correct |
14 |
Correct |
10 ms |
1620 KB |
Output is correct |
15 |
Correct |
17 ms |
2536 KB |
Output is correct |
16 |
Correct |
18 ms |
2476 KB |
Output is correct |
17 |
Correct |
20 ms |
2900 KB |
Output is correct |
18 |
Correct |
18 ms |
2908 KB |
Output is correct |
19 |
Correct |
23 ms |
3164 KB |
Output is correct |
20 |
Correct |
22 ms |
3156 KB |
Output is correct |