题解：P10455 Genius Acm

天才 Acm

一眼倍增。

我们可以定义一个左端点 $l=1$ 和右端点 $r=l$，然后二分一个 $p$，初始值为 $1$，如果 $[l,r+p]\text{这个区间的校验值}\le T$ 那么 $r=r+p,p=p\ast 2$ 否则 $p/=2$ 直到 $p=0$ 为止，就找到了从 $l$ 开始校验值不超过 $T$ 最长能到达的 $r$，这样就可以将 $ans++$，将 $l$ 重新设为 $r$。

重复以上过程，直到将整个数组分割完成为止。

而我们可以写一个 check 来判断 $[l,mid-1]$ 和 $[mid,r]$ 合并之后的校验值是否小于等于 $T$，而这两个区间很明显是有序的，我们只需要合并这两个区间，在判断是否合法即可。

code

#include <bits/stdc++.h>
#define sq(a) (a)*(a)
using namespace std;
const int N = 5e5 + 5;
typedef long long ll;
int n, m, ans, cas;//a是保存排序后的数组，b是merge后的数组，rec是保存原序列 
ll t, rec[N], a[N], b[N];
void merge(int l, int mid, int r) {
	int i = l, j = mid + 1;
	for (int k = l; k <= r; k++) {
		if (j > r || i <= mid && a[i] <= a[j])
			b[k] = a[i++];
		else
			b[k] = a[j++];
	}
}
bool check(int l, int mid, int r) {
	//[l, mid - 1]是前一段 
	for (int k = mid; k <= r; k++) a[k] = rec[k];
	sort(a + mid, a + r + 1);
	merge(l, mid - 1, r);
	ll sum = 0;
	for (int k = 1; k <= ((r - l + 1) >> 1) && k <= m; k++) 
		sum += sq(b[r - k + 1] - b[l + k - 1]);
	if (sum <= t) {
		for (int k = l; k <= r; k++) a[k] = b[k];
		return 1;
	}
	else
		return 0;
}
void solve() {
	int p = 1, l = 1, r = 1;
	a[l] = rec[l];
	while (r <= n) {
		if (!p) {
			ans++;
			l = ++r;
			a[l] = rec[l];
			p = 1;
		}
		else if (r + p <= n && check(l, r + 1, r + p)) {
			r = r + p;
			p <<= 1;
		}
		else
			p >>= 1;
	}
}
int main() {
	scanf("%d", &cas);
	while (cas--) {
		scanf("%d%d%lld", &n, &m, &t); 
		for (int i = 1; i <= n; i++) scanf("%d", rec + i);
		ans = 0;
		solve();
		printf("%lld\n", ans);
	}
	return 0;
}