牛客练习赛83 C.集合操作

C.集合操作(思维,后缀)

题目传送门:

C.集合操作

题目:

牛客练习赛83 C.集合操作_第1张图片

思路:

数据量很大,操作数和每次减去的数<=1e18。那么如果模拟的话肯定妥妥的爆炸。我们比较容易可以想到如果很多个数在同一个p的区块时(这些数中的最大值和最小值的差小于p)那么将这里面的所有数同时减去p时,他们的相对大小是不变的。

那么我们就可以先把原来的数排个序,从后往前操作,将后面的数不断并到前面的区块上,当并到不能操作时再退出。剩下的操作再进行暴力模拟,剩下的操作数量必然小于n。所以时间复杂度为nlogn。

AC Code

#include
using namespace std;
typedef long long LL;
const int N=1e6+10;
LL n,k,d;
LL a[N];
LL cal(int x)   //分区
{
     
    if(x>=0) return x/d;
    return (x+1)/d-1;
}
int main()
{
     
    scanf("%lld%lld%lld",&n,&k,&d);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    sort(a+1,a+1+n);    //从小到大排序
    if(d==0)
    {
     
        for(int i=1;i<=n;i++) printf("%lld ",a[i]);
        return 0;
    }
    int now=1;
    for(int i=n;i>=2;i--)
    {
     
        if(cal(a[i])==cal(a[i-1])) continue;
        LL p=cal(a[i])-cal(a[i-1]);
        LL q=n-i+1;
        LL sum=p*q;
        if(k>sum)
        {
     
            k=k-sum;
            continue;
        }
        now=i;
        break;
    }
    int f=k/(n-now+1);
    k=k%(n-now+1);
    for(int i=now+1;i<=n;i++)
    {
     
        LL p=cal(a[i])-cal(a[now]);
        a[i]=a[i]-p*d-f*d;
    }
    a[now]=a[now]-f*d;
    sort(a+1,a+1+n);
    for(int i=n;i>=n-k+1;i--)
        a[i]=a[i]-d;
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++) printf("%lld ",a[i]);
    system("pause");
    return 0;
}

你可能感兴趣的:(牛客竞赛,思维,牛客练习赛)