[bzoj4709]柠檬

[bzoj4709]柠檬

---

首先可以确定全部取完一定是最优的，然后我们也不难想出一个方案取的si在所选的坐标两端时一定是最优的。

由此我们可以分别对于每个si相等的点更新答案

f[i]=f[j]+a[i]*(s[i]-s[j])^2    //条件(a[j+1]==f[i])
//a[i]为值(上文Si),s[i],s[j]表示前缀中a[i]出现次数

然后就可以斜率优化了。注意这道题只能在上凸壳上二分

• 代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5,M=1e4+5;
typedef long long ll;

struct node{
    ll x,y;//"x=2*a*s[j],y=f[i]+a*s[j]*s[j]
};

vector<node>st[M];
int u[M],v[M];;
ll f[N];
int num[M];

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=10000;i++){
        st[i].push_back((node){0,0});
    }
    f[0]=0;
    for(int s,i=1;i<=n;i++){
        scanf("%d",&s);
        num[s]++;
        ll sj=num[s]-1;
        ll x3=sj*s*2,y3=f[i-1]+s*sj*sj;
        while(v[s]>1){
            ll x1=st[s][v[s]].x,y1=st[s][v[s]].y;
            ll x2=st[s][v[s]-1].x,y2=st[s][v[s]-1].y;
            if( (y3-y1)*(x2-x1) < (y2-y1)*(x3-x1) ){
                st[s].pop_back();
                --v[s];
            }else break;
        }
        ll mn=s*1ll*num[s]*num[s];
        ll k=num[s];
        st[s].push_back((node){x3,y3});v[s]++;
        int l=1,r=v[s];
        while(l<r){
            int mid=(l+r)>>1;
            ll x1=st[s][mid].x,y1=st[s][mid].y;
            ll x2=st[s][mid+1].x,y2=st[s][mid+1].y;
            if(y1-k*x1 <= y2-k*x2)l=mid+1;
            else r=mid;
        }
        int mid=(l+r)>>1;
        ll x1=st[s][mid].x,y1=st[s][mid].y;
        f[i]=y1-k*x1+mn;
    }
    printf("%lld\n",f[n]);
}