题意:给定长度为nnn的自然数序列aaa和整数kkk,要求构造长度为nnn的自然数序列bbb,使得0≤bi≤ai,∑bi=k0\leq b_i\leq a_i,\sum b_i=k0≤bi≤ai,∑bi=k,且∑bi(ai−bi2)\sum b_i(a_i-b_i^2)∑bi(ai−bi2)最大化。
n≤105n\leq 10^5n≤105
要求最大化的是b(a−b2)b(a-b^2)b(a−b2),记为f(x)=x(a−x2)f(x)=x(a-x^2)f(x)=x(a−x2)
注意到xxx每增加111,f(x)f(x)f(x)会增加Δi(x)=ai−3x2−3x−1\Delta_i(x)= a_i-3x^2-3x-1Δi(x)=ai−3x2−3x−1
注意到对于每个iii,当x≥0x\geq 0x≥0时,Δi(x)\Delta_i(x)Δi(x)单调递减,题目所求就是所有数中的前kkk大之和,而每个Δi\Delta_iΔi被取的一定是前面若干个,所以可以套路性地维护当前的xix_ixi和一个堆,每次取堆顶让对应的xi+1x_i+1xi+1
这样复杂度是O(klogn)O(k\log n)O(klogn)无法通过
但是每次加上的Δi(x)\Delta_i(x)Δi(x)是单调递减的(废话),所以可以二分最小的d=Δi(x)d=\Delta_i(x)d=Δi(x),再二分出每个Δi\Delta_iΔi被选了多少个,判断∑xi<k\sum x_i<k∑xi<k
最后∑xi\sum x_i∑xi可能小于kkk,但每个位置最多只能+1+1+1,直接在当前基础上暴力选k−∑xik-\sum x_ik−∑xi个
复杂度O(nlog2V)O(n\log^2V)O(nlog2V)
#include <iostream>
#include <cctype>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define MAXN 100005
using namespace std;
typedef long long ll;
const ll INF=4e18;
int a[MAXN],b[MAXN],x[MAXN],p[MAXN],n;
ll k,sum;
inline ll delta(const int& a,const int& x){return x==a? -INF:a-3ll*x*x-3ll*x-1;}
inline bool cmp(int A,int B){return delta(a[A],x[A])>delta(a[B],x[B]);}
inline bool check(ll v)
{for (int i=1;i<=n;i++){if (delta(a[i],0)<v) x[i]=0;else{int l=0,r=a[i]-1,mid;while (l<r){mid=(l+r+1)>>1;if (delta(a[i],mid)<v) r=mid-1;else l=mid;}x[i]=l+1;}}sum=0;for (int i=1;i<=n;i++) sum+=x[i];if (sum>k) return false;for (int i=1;i<=n;i++) b[i]=x[i];return true;
}
int main()
{cin>>n>>k;for (int i=1;i<=n;i++) scanf("%d",&a[i]);ll l=-INF,r=INF,mid,ans;while (l<r){mid=(l+r)/2;if (check(mid)) ans=mid,r=mid-1;else l=mid+1;}check(ans);for (int i=1;i<=n;i++) p[i]=i;sort(p+1,p+n+1,cmp);for (int i=1;i<=k-sum;i++) ++b[p[i]];for (int i=1;i<=n;i++) printf("%d ",b[i]);return 0;
}