0601 Genius ACM 倍增

最新推荐文章于 2024-01-21 12:35:59 发布

原创最新推荐文章于 2024-01-21 12:35:59 发布 · 347 阅读

1 ·

CC 4.0 BY-SA版权

倍增专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种利用倍增法解决特定校验值问题的方法。通过将数列分成若干段，确保每段的校验值不超过给定阈值T，目标是最小化分段数量。文中详细解释了倍增法的实现过程，包括如何通过排序和归并来优化算法复杂度。

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

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

现在给定一个长度为 N 的数列 A 以及一个整数 T。我们要把 A 分成若干段，使得每一段的“校验值”都不超过 T。求最少需要分成几段。

题解：

这个倍增的方法好像很厉害啊。
配对的方法一定是令最大最小、次大次小这样配对，证明可以自己写一写式子。然后我们就要尽量让每一段最长。初始化L=R=1，p=1，然后每次看[L，R+p]这一段是否符合，若符合，则R+=p，p*=2，否则p/=2，p为0时的R即为答案。每次只用把新增的一段排序，然后再与原来的归并合并，复杂度可以达到 $O(nlogn)$ 。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define LL long long
const int Maxn=500010;
int n,m;LL k,a[Maxn],b[Maxn],c[Maxn],t[Maxn];int lt;
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%lld",&n,&m,&k);
        for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
        int p=1,l=1,r=1,ans=0,lb;
        while(1)
        {
            if(l==n+1)break;
            b[lb=1]=a[l];
            int R;
            while(p)
            {
                R=min(r+p,n);
                lt=0;
                for(int i=r+1;i<=R;i++)t[++lt]=a[i];
                sort(t+1,t+1+lt);
                int l2=1,l1=0;
                for(int i=1;i<=lb;i++)
                {
                    while(l2<=lt&&t[l2]<=b[i])c[++l1]=t[l2++];
                    c[++l1]=b[i];
                }
                while(l2<=lt)c[++l1]=t[l2++];
                LL s=0;
                bool flag=true;int cnt=0;
                for(int i=1;i<=l1/2;i++)
                {
                    cnt++;
                    s+=(c[l1-i+1]-c[i])*(c[l1-i+1]-c[i]);
                    if(s>k){flag=false;break;}
                    if(cnt==m)break;
                }
                if(flag)
                {
                    if(r+p>=n){r=n;break;}
                    r=R,R+=p,p<<=1;
                    lb=l1;
                    for(int i=1;i<=lb;i++)b[i]=c[i];
                }
                else p>>=1;
            }
            ans++;
            l=++r;p=1;
        }
        printf("%d\n",ans);
    }
}