牛客练习赛51

E、数列

theme:给定n与m，要求你用正整数填充n个元素，使得这n个元素的和<=m,定义喜爱度为i的个数使得2<=i<=n且a[i]=a[i-1]+1。

1<=n<=1e5,1<=m<=1e9

solution:构造题。考虑枚举分成几个段i，若某段的长度为len,则该段的数为1~len,可知喜爱度应为n-i,所以我们的目标是使得n-i越大，也就是i越小越好。同时要满足每段的和相加<=m.当段数确定时，可以推出从长度len到长度len+1和增加了len+1,设最大的长都为maxx，最小的为minn,则maxx+1，minn-1时和增加了maxx-minn+2,所以我们时maxx与minn尽量靠近，也就是每段的长度尽量相同是最优的。当段树为i时可知每段的长度为n/i,还剩下res=n%i个元素，我们把它分配给res段，这res段长度为len+1,剩下i-res段长度为len .

#include<bits/stdc++.h>
typedef long long ll;
#define far(i,s,n) for(int i=s;i<n;++i)
using namespace std;

ll getSum(int len)
{
    return (1LL+len)*len/2;
}

int main()
{
    int n;
    ll m;
    cin>>n>>m;
    ll all=getSum(n);
    if(all<=m)
    {
        far(i,1,n+1)
            printf("%d ",i);
        puts("");
        return 0;
    }
    far(i,1,n+1)
    {
        int len=n/i;
        int res=n%i;
        if(getSum(len+1)*res+getSum(len)*(i-res)<=m)
        {
            far(j,1,res+1)
                far(k,1,len+2)
                    printf("%d ",k);
            far(j,1,i-res+1)
                far(k,1,len+1)
                    printf("%d ",k);
            puts("");
            break;
        }
    }
}