二分搜索【POJ2018】

传送门：http://poj.org/problem?id=2018

集训做的题目，专题是二分搜索，可是就是没想到怎么二分qwq。

生搬硬凑的二分，复杂度竟然是(n^2)*(logn)，直接暴力的复杂度是n^2，然后就自闭了一个多小时。

关键是没想到怎么在O(n)的复杂度下判断二分的结果。

---

思路：二分答案，对于每一个答案(mid)，数组全部减去这个mid，然后判断二分的结果的时候只要看有没有一段区间的和是大于0的。用前缀和维护一下区间和。复杂度成功的降到了nlogn。

注意判断二分的时候找一个区间和最大的区间就行了，不需要关心区间有多长。

总之很巧妙就对了！

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const double eps = 1e-5;
const int maxn = 1e5+7;
double a[maxn];
double s[maxn];
double tmp[maxn];
int n,f;
inline bool check(double x)
{
	for(int i=1;i<=n;i++)
	{
		tmp[i] = a[i] - x;
		s[i] = s[i-1]+tmp[i];
	}
	double mx = -1e9+7;
	double mn = 1e9+7;
	for(int i=f;i<=n;i++)
	{
		mn = min(mn,s[i-f]);
		mx = max(s[i]-mn,mx);
	}
	if(mx>=0)
	{
		return true; 
	}
	else
	{
		return false;
	}
}
int main()
{
	while(cin>>n>>f)
	{
		for(int i=1;i<=n;i++)
		{
			scanf("%lf",a+i);
		}
		double l = -1e6;
		double r = 1e6;
		while(r-l>eps)
		{
			double mid = (l+r)/2.0;
			if(check(mid))
			{
				l = mid;
			}
			else
			{
				r = mid;
			}
		}
		cout<<int(r*1000)<<endl;
	}
	return 0;
}

PS：用l*1000输出结果WA了好久，怎么输出都是6499那一片，最后用r*1000成功AC。

PPS：刚开始用的cin 。TLE到怀疑人生，l和r怎么调都不对。换了scanf直接AC了qwq