题意:在数轴上有n个城市(由坐标给出),要建立m(m<n)个邮局,邮局建在城市上。问所有城市距离最近邮局的距离之和。
思路:dp[i][j]:在前i个村庄中建立j个邮局的最小耗费;dis[i][j]:在第i个村庄到第j个村庄中建立1个邮局的最小耗费。经过简单比较可知,dis[i][j]的求法为把邮局建立在i和j的中点村庄处(如果一共有偶数个村庄那么中间的两个村庄都可以)。于是转移方程:dp[i][j] = min(dp[i][j],dp[k][j-1]+dis[k+1][i])
DP的边界状态即为dp[i][1] = dis[1][i];
#include <stdio.h>
#include <string.h>
#define N 303
int n,m;
int dp[N][N],dis[N][N],s[N];
int min(int a,int b){
if(a<b)
return a;
return b;
}
int main(){
freopen("a.txt","r",stdin);
while(scanf("%d %d",&n,&m)!=EOF){
int i,j,k;
memset(dis,0,sizeof(dis));
for(i = 0;i<=n;i++)
for(j = 0;j<=n;j++)
dp[i][j] = 0x3fffffff;
for(i = 1;i<=n;i++)
scanf("%d",&s[i]);
for(k = 1;k<n;k++)//建立dis数组
for(i = 1;i+k<=n;i++){
dis[i][i+k] = s[i+k]-s[i];
for(j = 1;i+j<i+k-j;j++)
dis[i][i+k] += s[i+k-j] - s[i+j];
}
for(i = 1;i<=n;i++)
dp[i][1] = dis[1][i];
for(i = 1;i<=n;i++)
for(j = 2;j<=i&&j<=m;j++)
for(k = j-1;k<i;k++)
dp[i][j]=min(dp[i][j],dp[k][j-1]+dis[k+1][i]);
printf("%d\n",dp[n][m]);
}
return 0;
}