洛谷P3648 [APIO2014]序列分割【斜率优化】

最新推荐文章于 2024-09-16 08:34:05 发布

niiick

最新推荐文章于 2024-09-16 08:34:05 发布

阅读量161

点赞数

分类专栏：动态规划--斜率优化DP

本文链接：https://blog.csdn.net/niiick/article/details/88914691

版权

动态规划--斜率优化DP 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

时空限制 2000ms / 512MB

题目描述

你正在玩一个关于长度为 n 的非负整数序列的游戏。这个游戏中你需要把序列分成 k + 1 个非空的块。为了得到 k + 1块，你需要重复下面的操作 k次：

选择一个有超过一个元素的块（初始时你只有一块，即整个序列）

选择两个相邻元素把这个块从中间分开，得到两个非空的块。

每次操作后你将获得那两个新产生的块的元素和的乘积的分数。你想要最大化最后的总得分。

输入格式：

第一行包含两个整数 n 和 k。保证 $\leq n$
第二行包含 n个非负整数 $a_1, a_2, \cdots, a_n(0 \leq a_i \leq 10^4)$ ，表示前文所述的序列。

输出格式：

第一行输出你能获得的最大总得分。

第二行输出 k 个介于 1 到 n - 1之间的整数，表示为了使得总得分最大，你每次操作中分开两个块的位置。第 i 个整数 s_i 表示第 i 次操作将在 si 和 $s_{i + 1}$ 之间把块分开。

如果有多种方案使得总得分最大，输出任意一种方案即可

说明

限制与约定
第一个子任务共 11 分，满足 1≤k<n≤10。
第二个子任务共 11 分，满足 1≤k<n≤50。
第三个子任务共 11 分，满足 1≤k<n≤200。
第四个子任务共 17 分，满足 2≤n≤1000,1≤k≤min{n−1,200}。
第五个子任务共 21 分，满足 2≤n≤10000,1≤k≤min{n−1,200}。
第六个子任务共 29 分，满足 2≤n≤100000,1≤k≤min{n−1,200}

题目分析

一开始还想写区间DP，然后以看数据范围。。。
好久没写斜率优化了，有分不清凸包方向了。。还被卡了好久精度。。我好菜啊+_+

$d p [i] [k] = m i n (d p [j] [k - 1]) + s u m [j] * (s u m [i] - s u m [j])$
$sum[j]^2-dp[j][k-1]=sum[i]*sum[j]-dp[i][k]$

维护下凸壳

#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long lt;
typedef double dd;
#define sqr(x) ((x)*(x))

lt read()
{
    lt f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return x*f;
}

const int maxn=200010;
int n,K;
lt a[maxn],sum[maxn];
int ll,rr,q[maxn],pre[maxn][210];
lt dp[2][maxn];

void print(int x,int t)
{
    if(t!=1) print(pre[x][t-1],t-1);
    printf("%d ",x);
}

lt qans(int i,int j,int t){ return dp[(t-1)&1][j]+sum[j]*(sum[i]-sum[j]);}
dd calc(int a,int b,int t)
{
    if(sum[a]==sum[b]) return -1e18;
    dd tx=sum[b]-sum[a];
    dd ty=(sqr(sum[b])-dp[(t-1)&1][b])-(sqr(sum[a])-dp[(t-1)&1][a]);
    return ty/tx;
}

int main()
{
    n=read();K=read();
    for(int i=1;i<=n;++i) 
    a[i]=read(),sum[i]=sum[i-1]+a[i];
    
    for(int t=1;t<=K;++t)
    {
        ll=rr=1;
        for(int i=1;i<=n;++i)
        {
            while(ll<rr&&calc(q[ll],q[ll+1],t)<=sum[i]) ++ll;
            dp[t&1][i]=qans(i,q[ll],t); pre[i][t]=q[ll];
            while(ll<rr&&calc(q[rr-1],q[rr],t)>=calc(q[rr],i,t)) --rr;
            q[++rr]=i;
        }
    }
    
    printf("%lld\n",dp[K&1][n]);
    print(pre[n][K],K);
    return 0;
}