思路:
把方差展开,前缀记录。因为方差在数字接近时取最小。排序即可
注意:
交了好多次才过,初始的数字一定要大1e9都小,注意double 的精确度。
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef double db;
const int maxn = 10005;
ll mul[maxn];
ll a[maxn];
ll pre[maxn];
int main()
{
int n,m;
cin>>m>>n;
for(int i=1;i<=m;i++)
cin>>a[i];
sort(a+1,a+1+m);
for(int i=1;i<=m;i++)
{
mul[i]=mul[i-1]+a[i]*a[i];
pre[i]=pre[i-1]+a[i];
}
db res = 1e15+5;
for(int i=n;i<=m;i++)
{
db ans=0;
ans=(mul[i]-mul[i-n])-(pre[i]-pre[i-n])*(pre[i]-pre[i-n])*1.0/n;
res=min(res,ans);
}
printf("%lld",(ll) res );
}