第一次写斜率DP的题,参考了很多东西
经过这题对斜率dp的理解是:
斜率dp是dp的优化方法,目的是将不能得到最优解的状态舍弃,将可能得到最优解的状态保留
将dp方程的转化为不等式,通过数形结合的方法把每个状态的答案反应为坐标系上的斜率
通过单调的下标来维护一个斜率单调的可行解队列,像是维护一个凸图形
参考的题解:https://www.cnblogs.com/kuangbin/archive/2012/08/26/2657650.html
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
const int N=500000+10;
int dp[N];
int q[N],q0,q1;
int sum[N];
int n,m;
int getdp(int i,int j)
{
return dp[j]+m+(sum[i]-sum[j])*(sum[i]-sum[j]);
}
int getup(int j,int k)
{
return dp[j]+sum[j]*sum[j]-(dp[k]+sum[k]*sum[k]);
}
int getdown(int j,int k)
{
return 2*(sum[j]-sum[k]);
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
for(int i=1;i<=n;i++)
scanf("%d",&sum[i]);
sum[0]=dp[0]=0;
for(int i=1;i<=n;i++)
sum[i]+=sum[i-1];
q0=q1=0;
q[q1++]=0;
for(int i=1;i<=n;i++)
{
while(q0+1<q1&&getup(q[q0+1],q[q0])<=sum[i]*getdown(q[q0+1],q[q0]))
q0++;
dp[i]=getdp(i,q[q0]);
while(q0+1<q1&&getup(i,q[q1-1])*getdown(q[q1-1],q[q1-2])<=getup(q[q1-1],q[q1-2])*getdown(i,q[q1-1]))
q1--;
q[q1++]=i;
}
printf("%d\n",dp[n]);
}
return 0;
}