题目:
传送门
由方差的性质得出,方差最小值一定在连续有序的m项产出。
所以先对给出的数字按照从小到大的顺序排序。
然后化简公式。
因为方差s^2 = 1/n * [(x1-k)^2+(x2-k)^2+.......+(xn-k)^2];
题目要求ans=s^2*n,则
ans=[(x1-k)^2+(x2-k)^2+.......+(xn-k)^2];
ans=x1^2-2x1*k+k^2+x2^2-2x2*k+k^2+x3^2-2x3*k+k^2....+xn^2-2xn*k+k^2;
则:ans=(x1^2+x2^2+x3^2+x4^2+.....xn^2)-2k(x1+x2+x3+....+xn)+n*k^2;
因为:k=(x1+x2+x3+....+xn)/n,带入上式,得:
ans=(x1^2+x2^2+x3^2+x4^2+.....xn^2)-(x1+x2+x3+....+xn)^2/n;
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long ll;
const int maxn=1e4+5;
const ll INF=1e15+5;
int n,m;
ll sum1[maxn],sum2[maxn];
ll a[maxn];
double Min=(double)INF;
int main()
{
memset (sum1,0,sizeof(sum1));
memset (sum2,0,sizeof(sum2));
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
sort (a+1,a+n+1);
for (int i=1;i<=n;i++)
{
ll x=a[i]*a[i];
sum1[i]=sum1[i-1]+a[i];
sum2[i]=sum2[i-1]+x;
}
for (int i=m;i<=n;i++)
{
double temp=sum2[i]-sum2[i-m]-((sum1[i]-sum1[i-m])*(sum1[i]-sum1[i-m])*1.0/m);
Min=min(temp,Min);
}
printf("%lld\n",(ll)Min);
return 0;
}