码题杯 -拔河 题型:二分

86 阅读1分钟

码题集OJ-拔河 (matiji.net)

假设我们要找的是mid,那我们就所有数都减去mid,减完mid之后相加:

image.png

通过观察上图我们发现,mid越靠左,相加和越大。越靠右相加和越大。如果我们找到的mid得到的相加和大于0,那么我们就应该往右边找新mid,使相加和更小,这样恢复成原数组得到的相加和就越大。

同理,如果我们当前mid计算之后相加和小于0,那么我们就往左找新的Mid,使得相加和更大,这样恢复成原数组相加和也只会更大。

code

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
double a[N], b[N];
int n, m;
double l, r;

bool check(double mid) {
	double mn = 0;
	for (int i = 1; i <= n; i++) {
		b[i] = a[i] - mid + b[i - 1];
		if (i >= m) {
			mn = min(mn, b[i - m]);
			if (b[i] - mn >= 0)
				return true;
		}
	}
	return false;
}

int main() {

	cin >> n >> m;

	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		l = 1, r = max(r, a[i]);
	}


	while (r - l >= 1e-6) {
		double mid = (l + r) / 2;
		if (check(mid))
			l = mid;
		else
			r = mid;
	}


	cout << (int)(r * 1000) << endl;
	return 0;
}

image.png