BZOJ[3675][Apio2014]序列分割 斜率优化dp

传送门ber~

感谢Orz大佬斜率优化讲解
因为得分跟顺序无关,所以可以dp
f i , j 表示第 i 次在 j 处分割的最大得分, s i j = 1 j i a j ,则有

f i , j = m a x { f i 1 , k + ( s j s k ) ( s n s j ) }

状态 t k 优,当且仅当
f i 1 , t + ( s j s t ) ( s n s j ) > f i 1 , k + ( s j s k ) ( s n s j )

g j = f i 1 , j ,拆开括号,有
g t + s j s n s j 2 s n s t + s t s j > g k + s j s n s j 2 s n s k + s k s j
g t g k s n s t + s n s k > s j ( s k s t )
g t g k s n s t + s n s k s k s t > s j

再令 p j = g j s j s n ,那么
p t p k s t s k < s j

这个东西就是斜率啊!
将每个状态表示为 ( p i , s i ) 的点对,维护一个下凸壳,因为 s 是单调的,所以用单调队列搞搞就可以了

代码如下:

#include<algorithm>
#include<ctype.h>
#include<cstdio>
#define INF 2147483647000000ll
#define N 100050
using namespace std;
inline int read()
{
    int x=0,f=1;char c;
    do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
    do x=(x<<3)+(x<<1)+c-'0',c=getchar(); while(isdigit(c));
    return x*f;
}
typedef long long LL;
int n,k,l,r;
LL s[N],f[N],g[N],ans;
struct Point
{
    LL x,y;
    Point(){}
    Point(LL _,LL __):x(_),y(__){}
    double operator * (const Point &b)///算斜率
    {
        if(x==b.x) return y>b.y ?-INF:INF;
        return 1.0*(b.y-y)/(b.x-x);
    }
}q[N];
inline LL calc(int k)
{
    return g[k]-s[k]*s[n];
}
inline LL GetAns(int k)
{
    while(l<r && q[l].x+q[l].y*s[k]<q[l+1].x+q[l+1].y*s[k]) l++;
    return q[l].x-s[k]*s[k]+s[k]*s[n]+q[l].y*s[k];
}
inline void Add(Point k)
{
    while(l<r && q[r-1]*q[r]>q[r]*k) r--;
    q[++r]=k;
}
main()
{
    n=read();k=read();
    for(int i=1;i<=n;i++)
        s[i]=read()+s[i-1];
    for(int i=1;i<=k;i++)
    {
        l=r=1;
        for(int j=1;j<=n;j++)
        {
            f[j]=GetAns(j);
            Add(Point(calc(j),s[j]));
        }
        for(int j=1;j<=n;j++)
            g[j]=f[j];
    }
    for(int i=1;i<n;i++)
        ans=max(ans,g[i]);
    printf("%lld",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/waduan2/article/details/79721842