Post Office--邮局问题(动态规划)

Post Office

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.

Samples
Input 复制
10 5
1 2 3 6 7 9 11 22 44 50
Output
9

题目大意
有几个村庄,然后准备挑几个出来建邮局,村庄的位置在一个数轴上,要使村庄到离他们最近的邮局的总距离最短,想出方案,输出最短的距离。

思路:
假设n个村庄,建k个,那么我们转换成已经建成k-1个最小了,再在剩下的选一个建立最小距离和。

用两个数组来记录:
f[i][j] :是1-i个村庄,建立了j个邮局的最小消费
w[i][j]:是i-j村庄建立一个邮局的最小消费

这里求w数组优化需要来理解一下:
如果原本i和j之间有偶数个village,假设此时post office的修建地址设置为中位数中较大的数的位置,此时新加入i后,post office的位置不发生改变,那么仍然由如下计算公式:
w[i,j] =w[i,j-1]+a[i] - a[(j+i)/2]
·如果原本i和j之间有奇数个village(即(i+j)%2==0),新加入一个j之后,仍然可以取原来的K作为中位数,此时中位数位置不变,所以
w[i,j] = w[i,j-1]+a[i] - a[(j+i)/2]

这里附图理解一下
在这里插入图片描述

所以优化后的 w[i,j] = w[i,j-1]+a[i] - a[(j+i)/2]

最后最后的f数组转移方程:
dp[i][j]=min(dp[i][j],dp[k][j-1]+w[k+1][i]);

AC代码:

int a[maxn];
int w[400][400],dp[400][400];
int main(){
	int n = read() , m = read();
	for(int i=1;i<=n;i++) cin>>a[i];//输入位置,他本身就是前缀和
	for(int i=1;i<=n;i++){
		for(int j=1+i;j<=n;j++){
			w[i][j]=w[i][j-1]+a[j]-a[(i+j)/2];
		}
	} 
	memset(dp,0x3f3f3f3f,sizeof(dp));
	for(int i=1;i<=n;i++) dp[i][i]=0,dp[i][1]=w[1][i];//当只建立一个邮局时是相等的
	for(int i=1;i<=n;i++){
		for(int j=2;j<=min(m,i);j++){//邮局个数 
		//dp[i][j]=inf;
		for(int k=j;k<=i-1;k++){//枚举可转移的位置,当然比邮局个数大 
                    dp[i][j]=min(dp[i][j],dp[k][j-1]+w[k+1][i]);
                }
		}
	}
	cout<<n<<" "<<m<<" "<<dp[n][m]<<"\n"; 
	for(int i=1;i<=n;i++) cout<<a[i]<<" ";
}

不懂的可以直接问啊

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值