109. 天才ACM（倍增，归并）

先介绍一下倍增算法

倍增：

倍增和二进制划分两个思想相结合，快速幂就是采用了倍增的思想

有这样一个问题：

给定一个长度为N的数列A，然后进行若干次询问，每次给定一个整数T，求出最大的k,满足,T一定小于整个区间的和

线性探索的时间复杂度是O(N),而倍增可以优化到O(logn)

倍增原理：任何一个数都可以表示为多个2次幂的和

1.令p=1,k=0,sum=0

2.比较A数组中k之后的p个数的和与T的关系

• sum+S[k+p]-S[k]<=T  令sum+=S[k+p]-S[k]  k+=p,p*=2
• sum+S[k+p]-S[k]>T    令p/=2

3.重复上一步，知道p的值为0，此时的k就是答案

我们可以发现，p永远是2的整数幂，那么就符合定义，k可以表示为多个2的整数幂的和，最后一定能找到一个数，满足<=T且最大的条件，时间复杂度O(logn)

回到本题

对于一个集合S，要取S中最大的M个数和最小的M个数，贪心一下，最大和最小，次大和次小...

要使段数最小，就是要让一个段包含尽可能多的点，我们前面已经知道了线性探测的时间复杂度是O(n),双指针也是O(n),而倍增探测时间复杂度是O(logn)，不妨使用倍增探测，使每个段尽可能地大，就可以最终让段数最小

求段数的操作其实和上面的问题类似了，要注意的是还要求校验值，让一段区间内的校验值<=T,让这个区间尽可能大，所以引出了下面这个函数

ll get(int l,int r)
{//返回l，r内的校验值
    
    ll res=0;
    int k=0;
    for(int i=l;i<r;i++)
        t[k++]=a[i];
    sort(t,t+k);
    for(int i=0;i<m&&i<k;i++,k--)
        res+=sq(t[i]-t[k-1]);
    return res;
}

 这个函数的时间复杂度是O(nlogn)的，所以最终的时间复杂度是O(nlog^2n)

#include <iostream>
#include <algorithm>
using namespace std;

const int N=5e5+10;

typedef long long ll;
ll a[N];
ll t[N];
int n,m;
ll sq(ll x)
{//返回x^2
    return x*x;
}
ll get(int l,int r)
{//返回l，r内的校验值
    
    ll res=0;
    int k=0;
    for(int i=l;i<r;i++)
        t[k++]=a[i];
    sort(t,t+k);
    for(int i=0;i<m&&i<k;i++,k--)
        res+=sq(t[i]-t[k-1]);
    return res;
}
int main()
{
    int k;
    
    cin.tie(0),cout.tie(0);
    cin>>k;
    while(k--)
    {
        ll T;
        cin>>n>>m>>T;
        for(int i=0;i<n;i++)
            cin>>a[i];
        int res=0;  //段数
        
        int st=0,ed=0;
        while(ed<n)
        {//倍增,区间左闭右开
            int p=1;
            while(p)
            {   //由于左闭右开,可以取到n
                if(ed+p<=n&&get(st,ed+p)<=T)
                    ed+=p,p<<=1;  //p*=2,同时更新ed
                else
                    p>>=1;
            }
            st=ed;
            res++;
        }
        cout<<res<<'\n';
    }
}