POJ 1160 Post Office (区间dp,从n个村庄选m个点建邮局,使得每个村庄到邮局的距离和最小)

题意

给定n个村庄的坐标,选m个地方建邮局,要求各个村庄到其最近的邮局的距离之和最小。距离定义为坐标之差。

解题

设dp[i][j]表示前i个村庄建j个邮局的最小距离和。

先预处理对于每个区间建一个邮局的最小距离和。设dis[i][j]表示在区间[i][j]建一个邮局的最小距离和。dis[i][j]=dis[i][j-1]+a[j]-a[(i+j)/2].因为建一个邮局的话,显然建立在中位数上是最优的,画个图就明了了。多一个村庄,虽然邮局的位置可能变了,但是前i-1个距离之和是不变的。所以多出了的距离就是第j个村庄到邮局的距离。

考虑加一个邮局:
前k个村庄建j-1个邮局,在第k+1个村庄到第i个村庄之间建一个邮局。
dp[i][j]=max{dp[k][j-1]+dis[k+1][j]},k是决策。取值范围为[j-1,i-1].

AC代码

#include 
#include 
#include 
#include 
#define INF 0x3f3f3f3f
using namespace std;

const int maxn=330;
int dp[maxn][maxn];
int a[maxn];
int dis[maxn][maxn];//dis[i][j]表示在第i个村和第j个村之间建一个邮局的“代价”

int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        //预处理dis[i][j]
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
        {

            //显然在[i,j]放一个邮局,放在最中间是最优的
            dis[i][j]=dis[i][j-1]+a[j]-a[(i+j)/2];
        }
        for(int i=1;i<=n;i++)
        {
            dp[i][i]=0;
            dp[i][1]=dis[1][i];
        }
        for(int j=2;j<=m;j++)
        {
            for(int i=j+1;i<=n;i++)
            {
                dp[i][j]=INF;
                for(int k=j-1;k1]+dis[k+1][i]);
                }
            }
        }
        printf("%d\n",dp[n][m]);
    }
    return 0;
}

你可能感兴趣的:(动态规划)