QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#131894 | #5125. Adjusted Average | Yarema# | AC ✓ | 3831ms | 74112kb | C++17 | 2.7kb | 2023-07-28 19:20:10 | 2023-07-28 19:20:12 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define SZ(a) (int)a.size()
#define ALL(a) a.begin(), a.end()
#define FOR(i, a, b) for (int i = (a); i<(b); ++i)
#define RFOR(i, b, a) for (int i = (b)-1; i>=(a); --i)
#define MP make_pair
#define PB push_back
#define F first
#define S second
#define FILL(a, b) memset(a, b, sizeof(a))
typedef long long LL;
typedef pair<int, int> PII;
typedef vector<int> VI;
const int INF = 1e9 + 7;
bool bad(int a, int b, int c, int d)
{
return a == c || a == d || b == c || b == d;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
LL n, k, avg;
cin >> n >> k >> avg;
VI x(n);
set<pair<LL, PII>> b;
LL sum = 0;
FOR (i, 0, n)
{
cin >> x[i];
sum += x[i];
FOR (j, 0, i)
{
b.insert({x[i] + x[j], {i, j}});
}
}
long double ans = (long double)(abs(sum - avg * n)) / n;
if (k >= 1)
{
LL target = avg * (n - 1);
FOR (i, 0, n)
{
sum -= x[i];
ans = min(ans, (long double)(abs(sum - target)) / (n - 1));
sum += x[i];
}
}
if (k >= 2)
{
LL target = avg * (n - 2);
for (auto [val, p] : b)
{
sum -= val;
ans = min(ans, (long double)(abs(sum - target)) / (n - 2));
sum += val;
}
}
if (k >= 3)
{
LL target = avg * (n - 3);
FOR (i, 0, n)
{
sum -= x[i];
FOR (j, 0, i) b.erase({x[i] + x[j], {i, j}});
FOR (j, i + 1, n) b.erase({x[i] + x[j], {j, i}});
auto it = b.lower_bound({sum - target, {-1, -1}});
if (it != b.end())
{
sum -= it->first;
ans = min(ans, (long double)(abs(sum - target)) / (n - 3));
sum += it->first;
}
if (it != b.begin())
{
it--;
sum -= it->first;
ans = min(ans, (long double)(abs(sum - target)) / (n - 3));
sum += it->first;
}
FOR (j, 0, i) b.insert({x[i] + x[j], {i, j}});
FOR (j, i + 1, n) b.insert({x[i] + x[j], {j, i}});
sum += x[i];
}
}
if (k >= 4)
{
LL target = avg * (n - 4);
for (auto [val, p] : b)
{
int v1 = p.F, v2 = p.S;
sum -= val;
auto it = b.lower_bound({sum - target, {-1, -1}});
auto it2 = it;
while (it2 != b.end() && bad(v1, v2, it2->S.F, it2->S.S)) it2++;
if (it2 != b.end())
{
sum -= it2->first;
ans = min(ans, (long double)(abs(sum - target)) / (n - 4));
sum += it2->first;
}
it2 = it;
if (it2 != b.begin())
{
it2--;
while (it2 != b.begin() && bad(v1, v2, it2->S.F, it2->S.S)) it2--;
if (!bad(v1, v2, it2->S.F, it2->S.S))
{
sum -= it2->first;
ans = min(ans, (long double)(abs(sum - target)) / (n - 4));
sum += it2->first;
}
}
sum += val;
}
}
cout << fixed << setprecision(14) << ans << '\n';
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3836kb
input:
5 2 2 1 2 3 100 200
output:
0.00000000000000
result:
ok found '0.00000', expected '0.00000', error '-0.00000'
Test #2:
score: 0
Accepted
time: 1ms
memory: 3640kb
input:
5 4 -5 -6 -3 0 6 3
output:
0.50000000000000
result:
ok found '0.50000', expected '0.50000', error '0.00000'
Test #3:
score: 0
Accepted
time: 0ms
memory: 3700kb
input:
4 1 4 1 3 3 7
output:
0.33333333333333
result:
ok found '0.33333', expected '0.33333', error '0.00000'
Test #4:
score: 0
Accepted
time: 1ms
memory: 3700kb
input:
2 1 0 0 0
output:
0.00000000000000
result:
ok found '0.00000', expected '0.00000', error '-0.00000'
Test #5:
score: 0
Accepted
time: 1ms
memory: 3724kb
input:
2 1 1 2 3
output:
1.00000000000000
result:
ok found '1.00000', expected '1.00000', error '0.00000'
Test #6:
score: 0
Accepted
time: 1ms
memory: 3652kb
input:
4 1 0 -2 -1 1 2
output:
0.00000000000000
result:
ok found '0.00000', expected '0.00000', error '-0.00000'
Test #7:
score: 0
Accepted
time: 1ms
memory: 3728kb
input:
5 2 0 -1 2 3 -4 42
output:
0.00000000000000
result:
ok found '0.00000', expected '0.00000', error '-0.00000'
Test #8:
score: 0
Accepted
time: 1ms
memory: 3652kb
input:
6 2 0 -5 -1 1 3 3 5
output:
0.00000000000000
result:
ok found '0.00000', expected '0.00000', error '-0.00000'
Test #9:
score: 0
Accepted
time: 1ms
memory: 3832kb
input:
5 4 -1000000000 1000000000 1000000000 1000000000 1000000000 1000000000
output:
2000000000.00000000000000
result:
ok found '2000000000.00000', expected '2000000000.00000', error '0.00000'
Test #10:
score: 0
Accepted
time: 0ms
memory: 3600kb
input:
5 4 1000000000 -1000000000 -1000000000 -1000000000 -1000000000 -1000000000
output:
2000000000.00000000000000
result:
ok found '2000000000.00000', expected '2000000000.00000', error '0.00000'
Test #11:
score: 0
Accepted
time: 1ms
memory: 3748kb
input:
5 1 0 -16 -12 4 12 20
output:
1.00000000000000
result:
ok found '1.00000', expected '1.00000', error '0.00000'
Test #12:
score: 0
Accepted
time: 1ms
memory: 3644kb
input:
8 2 0 -18 -6 -6 0 0 24 36 42
output:
1.00000000000000
result:
ok found '1.00000', expected '1.00000', error '0.00000'
Test #13:
score: 0
Accepted
time: 1ms
memory: 3828kb
input:
5 1 0 10000 -10 -100 -1000 -3900
output:
998.00000000000000
result:
ok found '998.00000', expected '998.00000', error '0.00000'
Test #14:
score: 0
Accepted
time: 1ms
memory: 3648kb
input:
15 3 1 -3 3 0 -3 3 2 0 3 -1 -1 0 -3 1 -1 -2
output:
0.41666666666667
result:
ok found '0.41667', expected '0.41667', error '0.00000'
Test #15:
score: 0
Accepted
time: 1ms
memory: 3704kb
input:
6 1 0 -1 -1 -1 1 1 1
output:
0.00000000000000
result:
ok found '0.00000', expected '0.00000', error '-0.00000'
Test #16:
score: 0
Accepted
time: 1ms
memory: 3696kb
input:
8 1 80 -90 52 -98 -79 70 86 38 -77
output:
80.00000000000000
result:
ok found '80.00000', expected '80.00000', error '0.00000'
Test #17:
score: 0
Accepted
time: 1ms
memory: 3776kb
input:
5 3 37 3 72 88 23 -51
output:
0.50000000000000
result:
ok found '0.50000', expected '0.50000', error '0.00000'
Test #18:
score: 0
Accepted
time: 1ms
memory: 3652kb
input:
8 3 -50 -68 -98 34 55 -9 83 -64 -45
output:
1.80000000000000
result:
ok found '1.80000', expected '1.80000', error '0.00000'
Test #19:
score: 0
Accepted
time: 0ms
memory: 3664kb
input:
5 2 62 -21 82 9 65 94
output:
0.33333333333333
result:
ok found '0.33333', expected '0.33333', error '0.00000'
Test #20:
score: 0
Accepted
time: 1ms
memory: 3648kb
input:
10 4 -16 57 60 25 -60 -31 -70 25 -45 -8 -93
output:
0.00000000000000
result:
ok found '0.00000', expected '0.00000', error '-0.00000'
Test #21:
score: 0
Accepted
time: 1163ms
memory: 34996kb
input:
1000 4 -589682264 239898913 202060748 -888127376 -880309095 124799170 -494539478 -745680608 726341697 992941811 -992809843 -977069412 -627162219 -243367314 325137914 647494705 994977955 32926705 618107495 702088915 19011397 546487463 811676912 -599600278 178857199 475002521 -328010873 568834738 -650...
output:
597654934.96787148597650
result:
ok found '597654934.96787', expected '597654934.96787', error '0.00000'
Test #22:
score: 0
Accepted
time: 1160ms
memory: 34824kb
input:
1000 4 -520031192 -828928297 -939974056 -890063527 -821483207 202331771 452507962 -984471175 730065137 -788464067 -814085288 -308486435 -251817293 982579015 -779352535 417038541 664276772 822475842 509497915 -536596860 616913996 358016799 -839048180 358771449 -538531644 562815966 635875550 337992251...
output:
558582928.13755020080134
result:
ok found '558582928.13755', expected '558582928.13755', error '0.00000'
Test #23:
score: 0
Accepted
time: 0ms
memory: 3748kb
input:
5 3 78 -68 42 -52 42 50
output:
32.00000000000000
result:
ok found '32.00000', expected '32.00000', error '0.00000'
Test #24:
score: 0
Accepted
time: 1ms
memory: 3780kb
input:
5 4 0 11 7 -13 -3 -6
output:
0.25000000000000
result:
ok found '0.25000', expected '0.25000', error '0.00000'
Test #25:
score: 0
Accepted
time: 1ms
memory: 3740kb
input:
6 4 50483269 -812901056 614642221 945836657 -616198107 598529790 -614398526
output:
15984531.50000000000000
result:
ok found '15984531.50000', expected '15984531.50000', error '0.00000'
Test #26:
score: 0
Accepted
time: 1ms
memory: 3832kb
input:
5 4 4 8 20 -18 3 9
output:
0.33333333333333
result:
ok found '0.33333', expected '0.33333', error '0.00000'
Test #27:
score: 0
Accepted
time: 1ms
memory: 3648kb
input:
10 4 -472189650 -763683851 291512773 236754835 367611187 -650471393 -435169974 -783816043 -891608557 997972439 910540713
output:
36488132.50000000000000
result:
ok found '36488132.50000', expected '36488132.50000', error '0.00000'
Test #28:
score: 0
Accepted
time: 1ms
memory: 3648kb
input:
6 4 -286216652 -146841360 -702906210 328658776 -159898481 154871206 -736124545
output:
2794288.00000000000000
result:
ok found '2794288.00000', expected '2794288.00000', error '0.00000'
Test #29:
score: 0
Accepted
time: 2109ms
memory: 74000kb
input:
1500 3 1 -3 -1 3 -1 1 -3 0 -2 -2 1 0 0 2 1 2 0 -2 -3 0 3 3 1 -2 0 1 2 0 0 -1 3 1 2 -1 0 -3 0 0 -2 -1 -3 2 3 0 0 3 -3 -3 -3 0 -3 0 2 -2 -2 3 -1 2 3 1 -2 -1 -2 -1 0 0 -2 2 2 2 -2 2 -1 3 -2 2 -3 -3 2 -2 3 -2 -1 2 -1 1 1 0 0 3 -3 3 -2 -2 -3 3 0 0 -3 2 -1 -2 1 2 0 0 3 0 1 3 2 -1 3 -1 1 -3 1 -2 1 -1 3 2 0...
output:
0.97194388777555
result:
ok found '0.97194', expected '0.97194', error '0.00000'
Test #30:
score: 0
Accepted
time: 793ms
memory: 34816kb
input:
1000 4 0 -5 3 4 4 2 -2 -1 -3 -4 -1 2 -2 1 -3 4 0 -1 -5 -2 2 2 4 -2 -1 -2 -1 -2 -4 1 -1 3 -4 2 -1 -3 2 5 2 -1 -1 -1 -2 5 -5 0 -1 -4 3 5 0 3 2 -4 5 -3 -2 5 -1 -1 -3 -3 2 2 3 -3 1 5 2 -4 4 1 5 -5 -4 5 -2 -4 2 -4 1 0 -5 1 -4 1 1 -2 -4 5 4 -2 -4 1 -4 -5 3 -1 5 -4 -4 -5 -1 -2 -2 -1 -2 -5 2 5 3 -4 -1 1 3 5...
output:
0.00903614457831
result:
ok found '0.00904', expected '0.00904', error '0.00000'
Test #31:
score: 0
Accepted
time: 1ms
memory: 3828kb
input:
10 4 -652315249 248814082 -310330977 -7639307 -808957195 -937992709 -461255220 526406959 317472565 958347317 -387890815
output:
166637545.16666666667152
result:
ok found '166637545.16667', expected '166637545.16667', error '0.00000'
Test #32:
score: 0
Accepted
time: 1ms
memory: 3768kb
input:
6 4 802938834 279940040 -341612005 -603412852 781046065 -36716272 116954057
output:
272445781.50000000000000
result:
ok found '272445781.50000', expected '272445781.50000', error '0.00000'
Test #33:
score: 0
Accepted
time: 1ms
memory: 3564kb
input:
20 4 221873446 849467103 884117952 312351432 -736878277 -296631326 -653303218 924111525 705904846 927004812 -810297249 860376579 744928549 -85282366 -877593960 698190037 840265628 -392839463 828364237 -422208014 198779260
output:
17181.16666666666667
result:
ok found '17181.16667', expected '17181.16667', error '0.00000'
Test #34:
score: 0
Accepted
time: 0ms
memory: 3728kb
input:
15 4 275628638 -847206183 782559411 141146580 425437384 810109824 -265811904 668678866 900016918 -401762002 358260741 574537650 -352457220 -504612680 418460859 909853096
output:
125105.00000000000000
result:
ok found '125105.00000', expected '125105.00000', error '0.00000'
Test #35:
score: 0
Accepted
time: 3831ms
memory: 74112kb
input:
1500 4 -196602262 182496385 408694457 -222624009 214805955 -162557649 975365037 991056372 -575198924 720929271 558986749 384198211 379596031 569475572 855820636 -211385054 -369267874 -628871496 -385283148 87771320 470431396 855795002 -140541095 309235940 445547104 -148900802 -201853913 97844726 1545...
output:
188924087.61229946523963
result:
ok found '188924087.61230', expected '188924087.61230', error '0.00000'
Test #36:
score: 0
Accepted
time: 3809ms
memory: 73936kb
input:
1500 4 -758396603 -319345413 -730907113 -450973444 -109298520 585578419 -421003783 50740431 195234867 -362795494 -576401840 185218667 -930428757 -984662611 115977237 -346656567 -953312444 621728277 -431334687 -802216908 -245752212 571253732 53024151 417955390 -912781318 -456601409 -1114596 -97580521...
output:
751982161.70387700537685
result:
ok found '751982161.70388', expected '751982161.70388', error '0.00000'