bzoj 4518 征途斜率优化

最新推荐文章于 2019-11-05 22:53:27 发布

Loi_a

最新推荐文章于 2019-11-05 22:53:27 发布

阅读量308

点赞数 2

分类专栏： ===DP及优化===

本文链接：https://blog.csdn.net/Loi_a/article/details/52416987

版权

===DP及优化=== 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

我想写题解，可我语文不好。

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int sum[3002],f[3002][3002];
int n,m,head,tail,a;
int q[3002];
double get(int i,int x,int y)
{
    return ((double)(f[i][x]+sum[x]*sum[x]-f[i][y]-sum[y]*sum[y]))/((double)(sum[x]-sum[y]));
}
int main()
{
    for(int i=0;i<=3000;i++)
    for(int j=0;j<=3000;j++)
        f[i][j]=1000000000;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a);
        sum[i]=sum[i-1]+a;
    }
    f[0][0]=0;
    for(int i=1;i<=m;i++)
    {
        head=tail=1;
        for(int j=1;j<=n;j++)
        {
            while(head<tail&&get(i-1,q[tail],q[tail-1])>get(i-1,j,q[tail])) tail--;
            q[++tail]=j;
            while(head<tail&&get(i-1,q[head+1],q[head])<(double)(sum[j]<<1)) head++;
            f[i][j]=(sum[j]-sum[q[head]])*(sum[j]-sum[q[head]])+f[i-1][q[head]];
        }
    }
    printf("%d\n",f[m][n]*m-sum[n]*sum[n]);
    return 0;
}