pku1160 Post Office

题目链接：http://acm.pku.edu.cn/JudgeOnline/problem?id=1160

题意简述：有n个村庄，m个邮局，给定各村庄的坐标位置，求n个村庄建立m个邮局使得所有村庄距它最近邮局的距离的和最小，输出最小值。

分析：看起来和前一道题目有点类似，但是我写这道题目的时候感觉要麻烦一点，所以可以把这题归类为上一道题的变形加强。下面来确定状态，设dp[i][j]为前i个村庄建立j个邮局所得的最优值。

状态转移：在写转移方程之前还需对一些数值进行处理，相信我们都见过这个问题的一个特例：当邮局只有一个时，邮局的位置在所有村庄排列后的中位数对应的位置时，总距离最小。

所以这里同时可设cost[i][j]表示i—j段只有一个邮局时的最小总距离(所求方法已经给出)。所以现在写转移方程也变得容易了：dp[i][j]=min(dp[k][j-1]+cost[k+1][i])(1<=k<=i-1)

代码如下所示：

#include<iostream> #include<string.h> using namespace std; int a[301]; __int64 cost[301][301]; __int64 dp[301][31]; int main() { int v,p; while(cin>>v>>p) { for(int i=1;i<=v;i++) cin>>a[i]; for(int i=1;i<=v;i++) for(int j=1;j<=v;j++) cost[i][j]=999999999; for(int i=1;i<=v;i++) cost[i][i]=0; for(int i=1;i<=v;i++) for(int j=i+1;j<=v;j++) { int mid=i+(j-i+1)/2; int sum=0; for(int k=mid-1;k>=i;k--) sum+=a[mid]-a[k]; for(int k=mid+1;k<=j;k++) sum+=a[k]-a[mid]; cost[i][j]=sum; } memset(dp,0,sizeof(dp)); for(int i=2;i<=v;i++) { dp[i][1]=cost[1][i]; for(int j=2;j<=(i<p?i:p);j++) { dp[i][j]=999999999; for(int k=1;k<=i-1;k++) { int c=dp[k][j-1]+cost[k+1][i]; if(c<dp[i][j]) dp[i][j]=c; } } } cout<<dp[v][p]<<endl; } return 0; }