标签：tmp printf pow put tput space bbb ++ int
51Nod - 1098 最小方差
第1行：2个数M,N，(M > N, M <= 10000) 第2 - M + 1行：M个数的具体值(0 <= Xi <= 10000)
输出最小方差 * N的整数部分。
5 3 1 2 3 4 5
2
题解：
求 x(i+1) + x(i+2) + ... + x(i+2) 的平方差，
(x[i+1] - avg)^2 + (x[i+2] - avg)^2 + ... + (x[i+n] - avg)^2 = (x[i+1] ^ 2 + x[i+2]^2 + .. + x[i+n]^2 - 2*avg*(x[i+1] + ... + x[i+n]) + n*avg*avg )
根据这个可以将时间由O(n*m) 缩小到O(n)
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
using namespace std;
const int MAXN = 10000 + 10;
int n, m, num[MAXN], pow2[MAXN];
int cmp(const void *a, const void *b){
return (*(int *)a - *(int *)b);
}
int main(){
long long ans, sum_power, sum_num, tmp;
double avg;
while(scanf("%d %d", &m, &n) != EOF){
for(int i=0; i<m; ++i){
scanf("%d", &num[i]);
}
qsort(num, m, sizeof(num[0]), cmp);
for(int i=0; i<m; ++i){
pow2[i] = num[i] * num[i];
}
sum_num = 0;
sum_power = 0;
for(int i=0; i<n; ++i){
sum_num += num[i];
sum_power += pow2[i];
}
avg = sum_num *1.0 / n;
ans = (long long)(sum_power - 2*avg*sum_num + n*avg*avg);
for(int i=n; i<m; ++i){
sum_num += num[i] - num[i-n];
sum_power += pow2[i] - pow2[i-n];
avg = sum_num *1.0 / n;
tmp = (long long)(sum_power - 2*avg*sum_num + n*avg*avg);
if(tmp < ans){
ans = tmp;
}
}
printf("%lld\n", ans );
}
return 0;
}
标签：tmp printf pow put tput space bbb ++ int
原文地址：http://www.cnblogs.com/zhang-yd/p/6910639.html
