小度养猫

百度之星 2022 初赛一

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+10;
priority_queue<pair<int,int>>q;
int a[N];
signed main()
{
	int k,n,ans=0;
	scanf("%lld%lld",&n,&k);
	for(int i=1;i<=n;i++)
	scanf("%lld",&a[i]);
	for(int i=1;i<=k;i++)
	q.push(make_pair(a[i],i));
	for(int i=k+1;i<=n+k;i++)//把当前时间可以喂的所有猫放进去 喂最大的那个就可以了 
	{
		if(i<=n)
		q.push(make_pair(a[i],i));
		pair<int,int> x=q.top();
		q.pop();
		ans+=(i*i-x.second*x.second)*x.first;
	}
	printf("%lld\n",ans);
	return 0;
}