【ACWing】109. 天才ACM（配数学证明）

题目地址：

https://www.acwing.com/problem/content/111/

给定一个整数$M$，对于任意一个整数集合$S$，定义“校验值”如下：从集合$S$中取出$M$对数(即$2\times M$个数，不能重复使用集合中的数，如果$S$中的整数不够$M$对，则取到不能取为止)，使得“每对数的差的平方”之和最大，这个最大值就称为集合$S$的“校验值”。现在给定一个长度为$N$的数列$A$以及一个整数$T$。我们要把$A$分成若干段，使得每一段的“校验值”都不超过$T$。求最少需要分成几段。

输入格式：
第一行输入整数$K$，代表有$K$组测试数据。
对于每组测试数据，第一行包含三个整数$N,M,T$。
第二行包含$N$个整数，表示数列$A_1,A_2\dots A_N$。

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

数据范围：
$1\le K\le 12$
$1\le N,M\le 500000$
$0\le T\le 10^{18}$
$0\le A_i\le 2^{20}$

先证明一个定理，设$a_1\le a_2\le ...\le a_n$，$n$是偶数，设$f_n(a_1,...,a_n)$是某种配对方式下的差的平方和，我们证明当配对方式是$(a_1,a_n),(a_2,a_{n-1}),...$的情况下$f_n$取到最大。

反证法：当$n=2$的时候可以暴力验证，结论正确。对于$n$个数的情况，定理中要求$a_1$和$a_n$配对，如果不是如此，设$a_1<a_i<a_j<a_n$，而$a_1$不和$a_n$配对，要么$(a_1,a_i),(a_j,a_n)$，要么$(a_1,a_j),(a_i,a_n)$，无论怎样，都可以暴力验证将这四个数按照定理中的配对方式能得到更大的和，这就矛盾了。

接下来考虑求解原题。我们用倍增的方式来求解从$l$开始的最长的一段，使得那一段的校验值小于等于$T$，截取出这样的一段之后，再在剩余的数里继续进行这个操作（由于如果集合$A$真包含于集合$B$，那么$A$的校验值一定不超过$B$的校验值，所以每次取的时候要尽可能的多取数，那样更优）。

具体的倍增的方法是，对于左端点$l$，我们先设右端点$r=l$，设$p=1$，代表步长。接着看$[l,r+p]$这一段的校验值是否满足条件，如果满足，则延伸右端点并倍增步长，令$r$等于$r+p$，令$p$为两倍；如果不满足，则将$p$减半。如果$p=1$的时候仍然不满足，则$p$减半后会变成$0$，此时退出循环，$r$即为本次取到的最右的右端点。

在判断校验值是否满足条件的时候，需要对区间排序。然而我们每次是新加了一些数，前面已经加的数已经排好序了，从而用归并排序更加节省时间。具体看代码，代码如下：

#include <algorithm>
#include <cmath>
#include <iostream>
using namespace std;
using ll = long long;

const int N = 5e5 + 10;
int k;
int n, m;
ll t;
int a[N], b[N], tmp[N];

// 每次进check的时候，b[l:mid-1]都已经与a[l:mid-1]的数字相同，且排好序了
bool check(int l, int mid, int r) {
  for (int i = mid; i <= r; i++) b[i] = a[i];
  // 只排序后半部分，并用归并的方式将[l:r]的最终排序结果放入tmp中
  sort(b + mid, b + r + 1);
  int k = l, i = l, j = mid;
  while (i < mid && j <= r)
    if (b[i] <= b[j])
      tmp[k++] = b[i++];
    else
      tmp[k++] = b[j++];
  while (i < mid) tmp[k++] = b[i++];
  while (j <= r) tmp[k++] = b[j++];

  // 验证一下校验码
  ll sum = 0;
  for (int i = l, j = r; i < j && i < l + m; i++, j--) {
    ll diff = tmp[i] - tmp[j];
    sum += diff * diff;
    if (sum > t) return false;
  }
  // 如果本轮校验码符合要求，那么后面只会新加数进来，
  // 当前[l:r]的数不会变了，从而可以将排序结果填入b中
  for (int i = l; i <= r; i++) b[i] = tmp[i];
  return true;
}

int work() {
  int res = 0;
  int l = 1;
  b[l] = a[l];
  while (l <= n) {
    int p = 1, r = l;
    while (p) {
      // b[l:r]排好序
      if (r + p <= n && check(l, r + 1, r + p))
        r += p, p <<= 1;
      else
        p >>= 1;
    }

    l = r + 1;
    res++;
  }

  return res;
}

int main() {
  scanf("%d", &k);
  while (k--) {
    scanf("%d%d%lld", &n, &m, &t);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    printf("%d\n", work());
  }
}

时间复杂度$O(n\log n)$，空间$O(n)$。