最佳牛围栏 二分+前缀和

题目链接
平均值技巧,同时减去avg
思路:将最优问题转化为判定问题

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
const int MAX=0x3f3f3f3f;
const int N=100010;
int a[N];
double sum[N];
int n,m;
bool check(double avg){
	for(int i=1;i<=n;i++) sum[i]=sum[i-1]+a[i]-avg;
	
	double  min_a=MAX;
	for(int i=0,j=m;j<=n;i++,j++){  //i到j 的和为sum[j]-sum[i-1] ,所以i从0开始 
		min_a=min(min_a,sum[i]);
		if(sum[j]>=min_a) return true;
		
	}
	return false;
}
int main(){

	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	double l=0,r=2000;    
	while(r-l>1e-5){
		double mid=(l+r)/2;
		if(check(mid)) l=mid;
		else r=mid;
	}
	printf("%d\n",int(r*1000));
	return 0;
}

你可能感兴趣的:(二分,前缀和)