天才 Acm
一眼倍增。
我们可以定义一个左端点 l = 1 l=1 l=1 和右端点 r = l r=l r=l,然后二分一个 p p p,初始值为 1 1 1,如果 [ l , r + p ] 这个区间的校验值 ≤ T [l,r+p] \text{这个区间的校验值} \le T [l,r+p]这个区间的校验值≤T 那么 r = r + p , p = p ∗ 2 r=r+p,p=p \ast 2 r=r+p,p=p∗2 否则 p / = 2 p/=2 p/=2 直到 p = 0 p=0 p=0 为止,就找到了从 l l l 开始校验值不超过 T T T 最长能到达的 r r r,这样就可以将 a n s + + ans++ ans++,将 l l l 重新设为 r r r。
重复以上过程,直到将整个数组分割完成为止。
而我们可以写一个 check 来判断 [ l , m i d − 1 ] [l,mid-1] [l,mid−1] 和 [ m i d , r ] [mid,r] [mid,r] 合并之后的校验值是否小于等于 T T 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;
}