//二分加前缀和
//正确性:如何使方差最小呢用感性的思维 猜测一段连续的且有序的方差是最小的
//单调性:我们想到只有往数组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;
}
成绩统计(第十五届蓝桥杯c++A组)
最新推荐文章于 2024-09-03 10:38:37 发布