[暑假集训Day4T1]羊圈
ZYC同志开农场了?????
二分答案。
对于每一个二分出来的答案对其进行检查(check),检查是否有一个长度大于m的字段和的平均值大于mid。方法如下:先把原数组的每一个元素减去mid,储存进一个新的数组中,我们只需要看这个数组中有没有一段子串和大于0即可(因为都减去了mid,如果大于零加回来后平均值就大于mid了),我们通过一个变量val记录当需要计算的数为i时,前i-m个数的总和最小值(需要的是最大值)。前缀和O(n)统计最大字段和即可。
参考代码如下:
1 #pragma GCC optimize(3,"Ofast","inline") 2 #include<iostream> 3 #include<cstdio> 4 #define M 100005 5 #define int long long 6 #define N 100005 7 using namespace std; 8 int read() 9 { 10 int x=0,f=1;char ch=getchar(); 11 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} 12 while(ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();} 13 return x*f; 14 } 15 int n,m,a[N],f[N],prefix[N],ans,Min=21374404,Max; 16 bool check(int x) 17 { 18 int val=21374404,ans=-21374404; 19 for(int i=1;i<=n;i++)f[i]=a[i]-x; 20 for(int i=1;i<=n;i++)prefix[i]=prefix[i-1]+f[i]; 21 for(int i=m;i<=n;i++) 22 { 23 val=min(val,prefix[i-m]); 24 ans=max(ans,prefix[i]-val); 25 } 26 if(ans>=0)return 1; 27 else return 0; 28 } 29 signed main() 30 { 31 n=read();m=read(); 32 for(int i=1;i<=n;i++)a[i]=read(),a[i]*=1000,Min=min(Min,a[i]),Max=max(Max,a[i]); 33 int l=Min,r=Max; 34 while(l<=r) 35 { 36 int mid=l+r>>1; 37 if(check(mid))l=mid+1; 38 else r=mid-1; 39 } 40 cout<<l-1<<endl; 41 return 0; 42 }