4057 | 2023. 03. 12 11:09:38 | HorakZsofi | Sípálya (55 pont) | cpp11 | Elfogadva 55/55 | 82ms | 10708 KiB |
#include <bits/stdc++.h>
using namespace std;
int main() {
int n;
cin>>n;
int k;
cin>>k;
vector<long long> szm(n);
vector<long long>pref(n);
long long ossz=0;
long long megold=0;
long long megoldas=0;
map<long long, int>vmi; // szám -> hány darab
for(int i=0; i<n; i++){
int a;
cin>>a;
ossz+=a+i;
pref[i]=ossz;
szm[i]=a+i;
}
for(int i=0; i<k; i++){
vmi[szm[i]] += 1;
}
auto it = vmi.end();
it--;
megoldas=(k*it->first)-pref[k-1];
for(int i=k; i<n; i++){
vmi[szm[i-k]] -= 1;
if(vmi[szm[i-k]] == 0) {
vmi.erase(szm[i-k]);
}
vmi[szm[i]]+=1;
auto it = vmi.end();
it--;
megold=(k*it->first)-pref[i]+pref[i-k];
if(megold<megoldas){
megoldas=megold;
}
}
cout<<megoldas;
}
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
base | 55/55 | ||||||
1 | Elfogadva | 0/0 | 3ms | 1808 KiB | |||
2 | Elfogadva | 0/0 | 3ms | 2060 KiB | |||
3 | Elfogadva | 2/2 | 3ms | 2292 KiB | |||
4 | Elfogadva | 2/2 | 3ms | 2444 KiB | |||
5 | Elfogadva | 2/2 | 3ms | 2532 KiB | |||
6 | Elfogadva | 2/2 | 3ms | 2764 KiB | |||
7 | Elfogadva | 3/3 | 3ms | 3108 KiB | |||
8 | Elfogadva | 1/1 | 8ms | 3396 KiB | |||
9 | Elfogadva | 1/1 | 8ms | 3720 KiB | |||
10 | Elfogadva | 1/1 | 8ms | 3936 KiB | |||
11 | Elfogadva | 1/1 | 6ms | 3748 KiB | |||
12 | Elfogadva | 1/1 | 6ms | 3884 KiB | |||
13 | Elfogadva | 1/1 | 6ms | 4244 KiB | |||
14 | Elfogadva | 2/2 | 6ms | 4452 KiB | |||
15 | Elfogadva | 2/2 | 6ms | 4392 KiB | |||
16 | Elfogadva | 2/2 | 78ms | 10236 KiB | |||
17 | Elfogadva | 2/2 | 76ms | 10184 KiB | |||
18 | Elfogadva | 2/2 | 76ms | 10444 KiB | |||
19 | Elfogadva | 3/3 | 79ms | 10420 KiB | |||
20 | Elfogadva | 2/2 | 81ms | 10444 KiB | |||
21 | Elfogadva | 2/2 | 78ms | 10348 KiB | |||
22 | Elfogadva | 2/2 | 78ms | 10296 KiB | |||
23 | Elfogadva | 2/2 | 81ms | 10548 KiB | |||
24 | Elfogadva | 2/2 | 82ms | 10632 KiB | |||
25 | Elfogadva | 2/2 | 81ms | 10572 KiB | |||
26 | Elfogadva | 2/2 | 82ms | 10572 KiB | |||
27 | Elfogadva | 2/2 | 79ms | 10568 KiB | |||
28 | Elfogadva | 3/3 | 79ms | 10708 KiB | |||
29 | Elfogadva | 3/3 | 78ms | 10632 KiB | |||
30 | Elfogadva | 3/3 | 82ms | 10572 KiB |