POJ 1160 Post Office (经典dp)

来源：互联网发布：c4.5算法python实现编辑：程序博客网时间：2024/05/18 02:38

Post Office

Time Limit: 1000MS Memory Limit: 10000KTotal Submissions: 13973 Accepted: 7536

Description

There is a straight highway with villages alongside the highway. The highway is represented as an integer axis, and the position of each village is identified with a single integer coordinate. There are no two villages in the same position. The distance between two positions is the absolute value of the difference of their integer coordinates.

Post offices will be built in some, but not necessarily all of the villages. A village and the post office in it have the same position. For building the post offices, their positions should be chosen so that the total sum of all distances between each village and its nearest post office is minimum.

You are to write a program which, given the positions of the villages and the number of post offices, computes the least possible sum of all distances between each village and its nearest post office.

Input

Your program is to read from standard input. The first line contains two integers: the first is the number of villages V, 1 <= V <= 300, and the second is the number of post offices P, 1 <= P <= 30, P <= V. The second line contains V integers in increasing order. These V integers are the positions of the villages. For each position X it holds that 1 <= X <= 10000.

Output

The first line contains one integer S, which is the sum of all distances between each village and its nearest post office.

Sample Input

10 51 2 3 6 7 9 11 22 44 50

Sample Output

Source

IOI 2000

【题目大意】：用数轴描述一条高速公路，有V个村庄，每一个村庄坐落在数轴的某个点上，需要选择P个村庄在其中建立邮局，要求每个村庄到最近邮局的距离和最小。

【题目分析】：经典DP

1、考虑在V个村庄中只建立【一个】邮局的情况，显然可以知道，将邮局建立在中间的那个村庄即可。也就是在a到b间建立一个邮局，若使消耗最小，则应该将邮局建立在（a+b)/2这个村庄上（可以通过画图知道）。

2、下面考虑建立【多个】邮局的问题，可以这样将该问题拆分为若干子问题，在前i个村庄中建立j个邮局的最短距离，是在前【k】个村庄中建立【j-1】个邮局的最短距离与在【k+1】到第i个邮局建立【一个】邮局的最短距离的和。而建立一个邮局我们在上面已经求出。

3、状态表示，由上面的讨论，可以开两个数组

dp[i][j]:在前i个村庄中建立j个邮局的最小耗费

sum[i][j]:在第i个村庄到第j个村庄中建立1个邮局的最小耗费

那么就有转移方程：dp[i][j] = min(dp[i][j],dp[k][j-1]+sum[k+1][i]) DP的边界状态即为dp[i][1] = sum[1][i]; 所要求的结果即为dp[village_num][post office_num];

4、然后就说说求sum数组的优化问题，可以假定有6个村庄，村庄的坐标已知分别为p1,p2,p3,p4,p5,p6;那么，如果要求sum[1][4]的话邮局需要建立在2或者3处,放在2处的消耗为p4-p2+p3-p2+p2-p1=p4-p2+p3-p1

放在3处的结果为p4-p3+p3-p2+p3-p1=p4+p3-p2-p1，可见，将邮局建在2处或3处是一样的。现在接着求sum[1][5],现在处于中点的村庄是3，那么1-4到3的距离和刚才已经求出了，即为sum[1][4],所以只需再加上5到3的距离即可。同样，求sum[1][6]的时候也可以用sum[1][5]加上6到中点的距离。所以有递推关系：sum[i][j] = sum[i][j-1] + p[j] -p[(i+j)/2]

体会：

1)个人觉得就是个递推的思想。。。

2)不能乱重复定义MAX这样的本来就又的宏定义，就因为这样还WA了几次。。。

代码：

#include<cstdio>#include<cstring>#include<iostream>#define minx(a,b) (a)>(b)?(b):(a)#define INF 10000000using namespace std;int dp[305][35];int sum[305][305];int pp[305];int main(){    int v,p,i,j,k;    while(~scanf("%d%d",&v,&p))    {        memset(sum,0,sizeof(sum));        for(i=1;i<=v;i++)            scanf("%d",&pp[i]);        for(i=1;i<v;i++)             //把每两个村庄间的距离算出来            for(j=i+1;j<=v;j++)                sum[i][j]=sum[i][j-1]+pp[j]-pp[(i+j)/2];        for(i=1;i<=v;i++)        {             dp[i][i]=0;             dp[i][1]=sum[1][i];        }        for(j=2;j<=p;j++)        {            for(i=j+1;i<=v;i++)            {                dp[i][j]=INF;                for(k=j-1;k<i;k++)                    dp[i][j]=minx(dp[i][j],dp[k][j-1]+sum[k+1][i]);            }        }        printf("%d\n",dp[v][p]);    }    return 0;}//不要乱定义MAX,它是系统内的一个宏，就因为这个WA了几次啊。。。