40572023-03-12 11:09:38HorakZsofiSípálya (55 pont)cpp11Elfogadva 55/5582ms10708 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ÖsszpontTesztVerdiktIdőMemória
base55/55
1Elfogadva0/03ms1808 KiB
2Elfogadva0/03ms2060 KiB
3Elfogadva2/23ms2292 KiB
4Elfogadva2/23ms2444 KiB
5Elfogadva2/23ms2532 KiB
6Elfogadva2/23ms2764 KiB
7Elfogadva3/33ms3108 KiB
8Elfogadva1/18ms3396 KiB
9Elfogadva1/18ms3720 KiB
10Elfogadva1/18ms3936 KiB
11Elfogadva1/16ms3748 KiB
12Elfogadva1/16ms3884 KiB
13Elfogadva1/16ms4244 KiB
14Elfogadva2/26ms4452 KiB
15Elfogadva2/26ms4392 KiB
16Elfogadva2/278ms10236 KiB
17Elfogadva2/276ms10184 KiB
18Elfogadva2/276ms10444 KiB
19Elfogadva3/379ms10420 KiB
20Elfogadva2/281ms10444 KiB
21Elfogadva2/278ms10348 KiB
22Elfogadva2/278ms10296 KiB
23Elfogadva2/281ms10548 KiB
24Elfogadva2/282ms10632 KiB
25Elfogadva2/281ms10572 KiB
26Elfogadva2/282ms10572 KiB
27Elfogadva2/279ms10568 KiB
28Elfogadva3/379ms10708 KiB
29Elfogadva3/378ms10632 KiB
30Elfogadva3/382ms10572 KiB