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; 
}