BZOJ3675【斜率优化】

最新推荐文章于 2020-07-10 22:34:35 发布

Lethelody

最新推荐文章于 2020-07-10 22:34:35 发布

阅读量568

点赞数

分类专栏：动态规划

本文链接：https://blog.csdn.net/lethelody/article/details/44781927

版权

动态规划专栏收录该内容

24 篇文章 0 订阅

订阅专栏

注意到判断斜率的时候.相等也要删除.

/* I will wait for you */
 
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iostream>
#include<fstream>
#include<vector>
#include<queue>
#include<deque>
#include<set>
#include<map>
#include<string>
 
typedef long long LL;
typedef unsigned long long ULL;
 
using namespace std;
 
const int maxn=100010;
const int maxm=210;
const int maxs=26;
const int INF=1<<29;
const int P=1000000007;
const double error=1e-9;
 
LL c[maxn],f[maxn][2],q[maxn];
 
LL h(int x,int y,int t)
{
    return (f[x][t]-c[x]*c[x])-(f[y][t]-c[y]*c[y]);
}
 
LL g(int x,int y)
{
    return c[x]-c[y];
}
 
int main()
{
    int n,k;scanf("%d%d",&n,&k);
     
    memset(c,0,sizeof(c)); 
     
    for(int i=1;i<=n;i++) cin>>c[i];
    for(int i=1;i<=n;i++) c[i]+=c[i-1];
 
    memset(f,0,sizeof(f));
    for(int t=1;t<=k;t++)
    {
        int head=1,tail=1;q[head]=t;
        for(int i=t+1;i<=n;i++)
        {
            while(tail>head&&h(q[head],q[head+1],~t&1)<=-c[i]*g(q[head],q[head+1])) head++;
            int p=q[head];f[i][t&1]=c[p]*(c[i]-c[p])+f[p][~t&1];
            while(tail>head&&h(q[tail-1],q[tail],~t&1)*g(q[tail],i)<=h(q[tail],i,~t&1)*g(q[tail-1],q[tail])) tail--;
            q[++tail]=i;
        }
    }
     
    cout<<f[n][k&1]<<'\n';
    return 0;
}