WHOJ-1603 Minimum Sum
第五届华中区程序设计邀请赛暨武汉大学第十四届校赛网络预选赛的一道题。
思路:看到题目第一感觉就是排序然后选择连续的m个数,突然看到选择的数下标要递增,唉怎么办!五分钟后想起重新按下标排序就是最后的结果序列,所以下标要求并没有什么用(脑子短路五分钟),按小到大排序后,怎么求连续m个数之间的值呢?再加个前缀和数组sum就好了,第i个数的贡献就是a[i]*(i-1)-sum[i-1]。求完第一个后怎么转移到下一个,其实就是删除第一个数和后m-1个的贡献,加上新增的数和后m-1个数的贡献,然后就木有然后了!
题目链接
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int main(){
int n,m,a[100009],sum[100009];
int i;
while(~scanf("%d%d",&n,&m)){
for(i=1;i<=n;i++){
scanf("%d",&a[i]);
}
sort(a+1,a+1+n);
if(m==1){
printf("0\n");
continue;
}
int ans=0;
sum[1]=a[1];
for(i=2;i<=m;i++){
ans+=a[i]*(i-1)-sum[i-1];
sum[i]=sum[i-1]+a[i];
}
int nu=ans;
for(;i<=n;i++){
ans-=sum[i-1]-sum[i-m]-(m-1)*a[i-m];
ans+=a[i]*(m-1)-sum[i-1]+sum[i-m];
sum[i]=sum[i-1]+a[i];
nu=nu<ans?nu:ans;
}
printf("%d\n",nu);
}
return 0;
}