1098最小方差

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
解题思路：首先说是方差最小的部分，那么排序之后的这部分肯定是连续的，这样就可以通过维护前缀和进行求解。
首先化简求值，（注意最后输出的值是方差乘以N），

图片有点问题不再修改自己查看点击。化简之后就可以进行查询。
用树状数组进行维护。

#include <stdio.h>
#include <string.h>
#include <string>
#include <algorithm>
#include <math.h>
using namespace std;
typedef long long LL;
const int Maxn = 1e5;
int m,n;
LL val[Maxn];
LL fang[Maxn];
int input[Maxn];
int lowbit(int x){
    return x&(-x);
}
void init(){
    memset(val,0,sizeof(val));
    memset(fang,0,sizeof(fang));
}
void valupdate(int x,int add){
    while(x<=m+5){
        val[x]+=add;
        x+=lowbit(x);
    }
}
void fangupdate(int x,int add){
    while(x<=m+5){
        fang[x]+=add*add;
        x+=lowbit(x);
    }
}
LL valquery(int x){
    LL temp=0;
    while(x){
        temp+=val[x];
        x-=lowbit(x);
    }
    return temp;
}
double longlong_double(LL a,LL b){
    double temp = a/b;
    double cha = a-b*temp;
    double add = cha/b;
    double re = temp+add;
    return re;
}
LL fangquery(int x){
    LL temp=0;
    while(x){
        temp+=fang[x];
        x-=lowbit(x);
    }
    return temp;
}
int main(){
    init();
    scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++){
        int temp;
        scanf("%d",&input[i]);
    }
    sort(input+1,input+m+1);
    for(int i=1;i<=m;i++){
        valupdate(i,input[i]);
        fangupdate(i,input[i]);
    }
    int i=0;
    if(n==1){
        printf("0\n");
        return 0;
    }
    int resu = 0;
    LL sum1;
    LL sum2;
    LL su;
    while(i+n<=m){
        if(!resu){
            sum1 = fangquery(i+n)-fangquery(i);
            sum2 = abs(valquery(i+n)-valquery(i));
            resu = 1;
            su = sum1*n-sum2*sum2;
            //printf("##%f\n",sum1-(double)sum2*sum2/n);
        }
        else{
            LL tp1 = fangquery(i+n)-fangquery(i);
            LL tp2 = valquery(i+n)-valquery(i);
            su = min(su,tp1*n-tp2*tp2);
            /*if((sum1-(double)sum2*sum2/n)<(tp1-(double)tp2*tp2/n));
            {
                sum1 = tp1;
                sum2 = tp2;
                printf("yes\n");
            }
            printf("##%f\n",tp1-(double)tp2*tp2/n);*/
        }
        i++;
    }
    printf("%lld\n",(su/n));
    //printf("%f\n",sum1-(double)sum2*sum2/n);
    //printf("%I64d  %I64d\n",sum1,sum2);
    return 0;
}