poj 1160 动态规划(一维城市建邮局)

题意:在数轴上有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;
}


你可能感兴趣的:(poj 1160 动态规划(一维城市建邮局))