109. 天才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 <iostream>
#include <algorithm>
#include <cstdio>

#define ll long long 

using namespace std;

const int N = 500010;
int n, m, mid;
ll k, a[N], b[N], c[N];

void merge(int l, int mid, int r)
{
    //if(r == n) break;
    int i = l, j = mid + 1;
    for(int k = l; k <= r; k++) // k = L
        if(j > r || (i <= mid && b[i] <= b[j])) c[k] = b[i++];
        else c[k] = b[j++];
     //賦值回數組b在下面
}
    


ll f(int l, int r) //求校驗值 貪心取集合s中最大的M個數和最小的M個數，最大和最小配成一對...
{
    if(r > n) r = n;
    int small = min(m, (r - l + 1) >> 1);//求最小值，長度（r - l + 1）/ 2
    for(int i = mid + 1; i <= r; i++) b[i] = a[i]; //mid + 1 = 2 從2開始幅值，所以前面b[1] = a[1]
    sort(b + mid + 1, b + r + 1);
    merge(l, mid, r);
    ll ans = 0; //至少分幾段
    for(int i = 0; i < small; i++)
        ans += (c[r-i] - c[l+i]) * (c[r-i] - c[l+i]); //每對數的差的平方之和
    return ans;
}

void Genius_ACM()
{
    cin >> n >> m >> k; 
    for(int i = 1; i <= n; i++) scanf("%lld", &a[i]);
    int ans = 0, l = 1, r = 1;
    mid = 1;
    b[1] = a[1];
    while(l <= n)
    {
        int p = 1; //初始化步長p = 1
        while(p)
        {
            ll num = f(l, r + p); // 求校驗值num
            if(num <= k) //不超過整數k
            {
                mid = r = min(r + p, n);
                for(int i = l; i <= r; i++) b[i] = c[i]; //int i = l 是L
                if(r == n) break;
                p <<= 1; // p = p * 2
            }
            else p >>= 1;
        }
        ans ++;
        l = r + 1;
    }
    cout << ans << endl;
}

int main()
{
    int t;
    cin >> t;
    while(t--) Genius_ACM();
    return 0;
}
 

《算法競賽進階指南》0.6倍增