天才ACM

给定一个整数 M，对于任意一个整数集合 S，定义“校验值”如下:

从集合 S 中取出 M 对数(即 2∗M 个数，不能重复使用集合中的数，如果 S 中的整数不够 M 对，则取到不能取为止)，使得“每对数的差的平方”之和最大，这个最大值就称为集合 S 的“校验值”。

现在给定一个长度为 N 的数列 A 以及一个整数 T。

我们要把 A 分成若干段，使得每一段的“校验值”都不超过 T。

求最少需要分成几段。

输入格式
第一行输入整数 K，代表有 K 组测试数据。

对于每组测试数据，第一行包含三个整数 N,M,T 。

第二行包含 N 个整数，表示数列A1,A2…AN。

输出格式
对于每组测试数据，输出其答案，每个答案占一行。

数据范围
1≤K≤12,
1≤N,M≤500000,
0≤T≤1018,
0≤Ai≤220

输入样例：
2
5 1 49
8 2 1 7 9
5 1 64
8 2 1 7 9
输出样例：
2
1

#include<bits/stdc++.h>

#define ll long long
using namespace std;
const int N = 5e5 + 10;
ll k, a[N], b[N], c[N];
int n, m, T;

ll f(int l, int w, int r) {
    if (r > n)r = n;
    int t = min(m, (r - l + 1) >> 1);
    for (int i = w + 1; i <= r; i++)b[i] = a[i];
    sort(b + w + 1, b + r + 1);
    for (int u = l, i = l, j = w + 1; u <= r; u++) {
        if (j > r || (i <= w && b[i] <= b[j]))c[u] = b[i++];
        else c[u] = b[j++];
    }
    ll res = 0;
    for (int i = 0; i < t; i++)
        res += (c[l + i] - c[r - i]) * (c[l + i] - c[r - i]);
    return res;
}

int main() {
    cin >> T;
    while (T--) {
        scanf("%d%d%lld", &n, &m, &k);
        for (int i = 1; i <= n; i++)
            scanf("%lld", &a[i]);
        int l = 1, r = 1, w = 1;
        int ans = 0;
        b[1] = a[1];
        while (l <= n) {
            int p = 1;
            while (p) {
                ll num = f(l, w, r + p);
                if (num <= k) {
                    w = r = min(r + p, n);
                    for (int i = l; i <= r; i++)b[i] = c[i];
                    if (r == n)break;
                    p <<= 1;
                } else p >>= 1;
            }
            ans++;
            l = r + 1;
        }
        printf("%d\n", ans);
    }
    return 0;
}