给n个村子建p个邮局,求最佳的建设方案使得每个村子到最近的邮局的距离和最短,输出最短距离。
首先递推求出n个村子建1个邮局的最佳方案,画下图很容易理解。
再递推求解多个邮局的最佳方案,具体的看注释吧。
/*
//k从1枚举到i-1就够了
dp[i][j]=min(dp[i][j],dp[k][j-1]+sum[k+1][i]);
这个画下图很好推出来的。
sum[i][j]=sum[i][j-1]+p[j]-p[(i+j)/2];
*/
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int sum[305][305]; //在i到j之间建一个邮局的最小消耗
int p[305];
int dp[305][305]; //前i个村子建j个邮局的最小消耗
int main()
{
int n,k;
while(~scanf("%d%d",&n,&k))
{
memset(dp,0x3f,sizeof(dp));
memset(sum,0,sizeof(sum));
p[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&p[i]);
}
for(int i=1;i<=n;i++)
{
for(int j=i+1;j<=n;j++)
{
sum[i][j]=sum[i][j-1]+p[j]-p[(i+j)/2];
}
dp[i][1]=sum[1][i];
dp[i][i]=0;
}
for(int j=2;j<=k;j++)
{
for(int i=j+1;i<=n;i++)
{
for(int k=1;k<i;k++)
{
dp[i][j]=min(dp[i][j],dp[k][j-1]+sum[k+1][i]);
}
}
}
printf("%d\n",dp[n][k]);
}
return 0;
}