AcWing 102. 最佳牛围栏(二分答案)

题意:给一个长度为n的序列,选取一段大于等于k的子序列,求可能子序列的最大平均值。
思路:首先答案是可以二分的,每次二分去判断是否有可行解。
然后就是怎么写二分的check了,区间的平均数大于当前所要判断的平均值等价于区间每个数减去平均值后区间和大于0,然后写个双指针,维护前序最小的前缀。

int n, len;
int a[N];
double sum[N];
bool check(double now)
{
	f(i, 1, n)sum[i] = sum[i - 1] + a[i] - now;//区间平均数大于mid等价于区间每个数减去mid后和大于0
	double mi = sum[0];
	for (int i = 0, j = len;j <= n;j++, i++)//维护前序最小前缀
	{
		mi = min(sum[i], mi);
		if (sum[j] - mi > 0)return true;
	}
	return false;
}
int main()
{
	cin >> n >> len;
	f(i, 1, n)scanf("%d", &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;
	}
	cout << (int)(r * 1000) << endl;
	return 0;
}

你可能感兴趣的:(基础算法,二分法)