NOIP提高组【JZOJ4790】选数问题

Description

在麦克雷的面前有N个数，以及一个R*C的矩阵。现在他的任务是从N个数中取出R*C个，并填入这个矩阵中。矩阵每一行的法值为本行最大值与最小值的差，而整个矩阵的法值为每一行的法值的最大值。现在，麦克雷想知道矩阵的最小法值是多少。

Data Constraint

30%：1<=n,r,c<=100
50%：1<=n,r,c<=1000
100%：1<=r,c<=10^4,r*c<=n<=5*10^5,0< pi<=10^9

Solution

我们发现显然是将排序后的序列取出连续的R段长度为C的数，而答案则为这R段中每一段头减尾的最大值。于是我们想到二分答案。对于每一次二分出来的答案x，我们计算每个点在相差不超过x的情况下最多能向左跳多远，这可以用单调队列O(N)实现。然后我们将每个数加入队列，从右往左做，对于每一个枚举的i,我们将能跳到i的左方但坐标在i右方的队列中的数统计一下，假设数量大于C，我们就将段数加1，并将队列左移c位，直至数量小于C，这也可以用单调队列O(N)实现。所以总时间复杂度为O(N$logN$)。

代码

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=500005;
int f[maxn],a[maxn],d[maxn],n,r1,c,i,t,j,k,l,x,y,z,r,mid,num;
int main(){
//  freopen("data.in","r",stdin);
    scanf("%d%d%d",&n,&r1,&c);
    for (i=1;i<=n;i++)
        scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    l=1;
    r=1000000000;
    while (l<r){
        mid=(l+r)/2;d[0]=d[1]=num=1;t=0;k=0;f[1]=1;
        for (i=2;i<=n;i++){
            while (a[i]-a[d[d[0]]]>mid && d[0]<=num)d[0]++;
            if (d[0]<=num) f[i]=d[d[0]];
            else f[i]=i;
            d[++num]=i;
        }
        for (i=1;i<=n;i++)
            d[i]=i;
        d[0]=n;
        for (i=n;i>=1;i--){
            while (f[d[d[0]]]>i && d[0]) d[0]--;
            while (d[0]-i+1>=c) t++,d[0]-=c;
        }
        t+=d[0]/c;
        if (t>=r1) r=mid;
        else l=mid+1;
    }
    printf("%d\n",l);
}