BZOJ 3675 APIO2014 序列分割 斜率优化DP

题意:链接

方法:斜率优化DP

解析:这题BZ的数据我也是跪了,特意去网上找到当年的数据后面二十个最大的点都过了,就是过不了BZ。

看到这道题自己第一发DP是这么推得:

设f[i][j]是第j次分第i个的最大得分。

那么会推出来 f[i][j]=max(f[k][j1]+sum[i k]sum[1 k1](sum[k i]sum[i+1 n])) 然后我发现这个式子的复杂度很高暂且不说,就光那个或的讨论就很费劲。

于是想了想就放弃了这个念头,中规中矩的去想。

按照以往的思路设出状态f[i][j]代表前i个分j次的最大得分。

能推出转移方程

f[i][j]=max(f[k][j1]+sum[k](sum[j]sum[k]))

之后对于样例手写一遍看出它的正确性后进行后面的讨论

我们发现如果n^2的枚举是肯定不行的,所以才去一种方式进行维护,因为有k的元素的存在,所以从斜率角度入手。

具体推导过程就不写了,得出的结果是:

f[j][tmp1]f[k][tmp1]+sum[k]2sum[j]2sum[k]sum[j]<=sum[i]

则说明k比j优。

所以尾部就是维护g[j,k]

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 100010
#define K 210
using namespace std;
typedef unsigned long long ll;
ll sum[N],a[N],f[N][2],q[N];
ll n,k;
int tmp;
ll fy(int j1,int j2,int d)
{
    return f[j1][d]-f[j2][d]+sum[j2]*sum[j2]-sum[j1]*sum[j1];
}
ll fx(int j1,int j2)
{
    return sum[j2]-sum[j1];
}
int main()
{
    scanf("%llu%llu",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%llu",&a[i]);
        sum[i]=sum[i-1]+a[i];
    }
    tmp=0;
    for(int j=1;j<=k;j++)
    {
        tmp^=1;
        int head=0,tail=0;
        q[head]=0;
        for(int i=1;i<=n;i++)
        {
            while(head<tail&&fy(q[head],q[head+1],tmp^1)<=fx(q[head],q[head+1])*sum[i])head++;
            while(head<tail&&fy(q[tail-1],q[tail],tmp^1)*fx(q[tail],i)>=fy(q[tail],i,tmp^1)*fx(q[tail-1],q[tail]))tail--;
            int t=q[head];
            f[i][tmp]=f[t][tmp^1]+sum[t]*(sum[i]-sum[t]);
            q[++tail]=i;
        }
    }
    printf("%llu\n",f[n][tmp]);
}

你可能感兴趣的:(数据,方法,序列,SUM,max)