【EOJ Monthly 2019.02 - F】方差(数学,前缀和,积的前缀和)

9 篇文章 0 订阅

题干:

单测试点时限: 2.0 秒

内存限制: 256 MB

“放弃不难,但坚持一定很酷。”

QQ 小方已经在体育馆苦练一天射箭了,但他还在坚持。

QQ 小方每天都要在朋友圈晒自己的训练记录。他一共进行了 n 次射箭,成绩分别是 x1,x2,⋯,xn 。为了表现自己的发挥十分稳定,QQ 小方决定选出其中的 m 次成绩,使得他们的方差是所有可以选择的方案中最小的。

对于 m 个元素组成的数列 a1,a2,⋯,am ,我们知道他们的方差 σ2=(a1−a¯)2+(a2−a¯)2+⋯+(am−a¯)2m ,其中 a¯=a1+a2+⋯+amm 。

但是这个问题对 QQ 小方来说太难了,你需要去帮助 QQ 小方。

为了方便,现在你需要输出这个最小的 σ2 乘以 m2 以后的结果。

输入

输入一行包含两个正整数 n (1≤n≤106 ) 和 m (1≤m≤n )。接下来一行包含 n 个整数 x1,x2,⋯,xn (1≤xi≤103 )。

输出

输出一行包含一个整数,表示答案。为了方便,我们需要输出最小的 σ2 乘以 m2 以后的结果。

样例

Input

5 3
1 2 3 4 5

Output

6

解题报告:

   跟这题一样【51nod - 1098】 最小方差,不解释了。

AC代码:

#include<cstdio>
#include<assert.h>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<set>
#include<string>
#include<cmath>
#include<cstring>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define pm make_pair
using namespace std;
const int MAX = 2e6 + 6;
const ll INF =9223372036854775807;
ll a[MAX],sqr[MAX],sum[MAX],Sqr,Sum,ans = INF;//ans = m*Sqr - Sum*Sum;
int main()
{
	int n,m;
	cin>>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++) sum[i] = sum[i-1] + a[i],sqr[i] = sqr[i-1] + a[i]*a[i];
	Sqr = sqr[m],Sum = sum[m];
	for(int l = 1; l+m-1<=n; l++) {
		int r = l + m - 1;
		Sqr = sqr[r] - sqr[l-1];
		Sum = sum[r] - sum[l-1];
		ans = min(ans,m*Sqr - Sum*Sum);
	}
	cout << ans << endl;
	return 0 ;
 } 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值