BZOJ 1705: [Usaco2007 Nov]Telephone Wire 架设电话线 DP_优化_推导
Code:
#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin)
#define Max(a,b) (a=a>b?a:b)
#define Min(a,b) (a=a<b?a:b)
using namespace std;
int f[103],h[100003],min1[103],min2[103];
int main()
{
// setIO("input");
int n,c,maxv=0;
scanf("%d%d",&n,&c);
for(int i=1;i<=n;++i)
scanf("%d",&h[i]),Max(maxv,h[i]);
memset(f,0x3f,sizeof(f));
for(int i=h[1];i<=maxv;++i) f[i]=(i-h[1])*(i-h[1]);
for(int i=2;i<=n;++i)
{
memset(min1,0x3f,sizeof(min1));
memset(min2,0x3f,sizeof(min2));
for(int j=h[i-1];j<=maxv;++j)
min1[j]=min(min1[j-1],f[j]-c*j);
for(int j=maxv;j>=1;--j)
min2[j]=min(min2[j+1],j>=h[i-1]?f[j]+c*j:0x3f3f3f3f);
for(int j=h[i];j<=maxv;++j)
f[j]=(j-h[i])*(j-h[i])+min(min1[j]+c*j,min2[j]-c*j);
}
int ans=0x3f3f3f3f;
for(int i=h[n];i<=maxv;++i) ans=min(ans,f[i]);
printf("%d\n",ans);
return 0;
}