题目描述:
1098 最小方差
若x1,x2,x3......xn的平均数为k。
则方差s^2 = 1/n * [(x1-k)^2+(x2-k)^2+.......+(xn-k)^2] 。
方差即偏离平方的均值,称为标准差或均方差,方差描述波动程度。
给出M个数,从中找出N个数,使这N个数方差最小。 收起
输入
第1行:2个数M,N,(M > N, M <= 10000) 第2 - M + 1行:M个数的具体值(0 <= Xi <= 10000)
输出
输出最小方差 * N的整数部分。
输入样例
5 3 1 2 3 4 5
输出样例
2
思路:
1e4的数据量,直接枚举每8个数即可,时间复杂度O(N*M)。
O(N)的做法,预处出a【i】^2的前缀和,a【i】的前缀和,用公式D(X)=E(X^2)-E(X)*E(X),可以避免O(M)求
(a【i】-ave)的平方。
代码实现:
#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int N=2e5+100;
LL a[N];
LL sum1[N],sum2[N];
int main()
{
int n,m;
while(cin>>n>>m)
{
for(int i=1;i<=n;i++)cin>>a[i];
LL ans=99999999999999999;
sort(a+1,a+1+n);
for(int i=1;i<=n;i++)sum1[i]=sum1[i-1]+(LL)a[i]*a[i];
for(int i=1;i<=n;i++)sum2[i]=sum2[i-1]+a[i];
for(int i=1;i+m-1<=n;i++)
{
LL tmp1=sum1[i+m-1]-sum1[i-1];
double tmp2=1.0*(sum2[i+m-1]-sum2[i-1])*(sum2[i+m-1]-sum2[i-1])/m;
double res=(double)tmp1-tmp2;
ans=min(ans,(LL)res);
}
cout<<ans<<endl;
}
return 0;
}
The end;