成绩统计(第十五届蓝桥杯c++A组)

//二分加前缀和
//正确性：如何使方差最小呢用感性的思维 猜测一段连续的且有序的方差是最小的 
//单调性：我们想到只有往数组b加个数才(有可能)使得存在连续k个数的方差减小 
//时间复杂度为O(n*log(n)*log(n)) 
#include<bits/stdc++.h>
using namespace std;
#define int long long 
const int N=1e5+5;
int a[N],b[N],pre[N],pre2[N];//pre为前缀和数组pre2为平方的前缀和数组 
int n,k,t;
int check(int x)
{
    for(int i=1;i<=x;i++)
    b[i]=a[i];
    sort(b+1,b+x+1);
    for(int i=1;i<=n;i++)
    {
        pre[i]=pre[i-1]+b[i];
        pre2[i]=pre2[i-1]+b[i]*b[i];
    }
    for(int i=1;i+k-1<=x;i++)
    {
        int ans=0;
        ans+=k*(pre2[i+k-1]-pre2[i-1])-(pre[i+k-1]-pre[i-1])*(pre[i+k-1]-pre[i-1]);//计算某个区间的方差/k*k 
        if(ans<k*k*t)
        return 1;
    }
    return 0;
}
signed main()
{
    cin>>n>>k>>t;
    for(int i=1;i<=n;i++)
    cin>>a[i];
    int l=1,r=n;
    while(l<r)
    {
        int mid=l+r>>1;
        if(check(mid))
        r=mid;
        else
        l=mid+1;
    }
    if(check(l))
    cout<<l;
    else
    cout<<-1;
}