2.6基本算法之动态规划 7624:山区建小学——区间DP

描述

政府在某山区修建了一条道路，恰好穿越总共m个村庄的每个村庄一次，没有回路或交叉，任意两个村庄只能通过这条路来往。已知任意两个相邻的村庄之间的距离为di（为正整数），其中，0 < i < m。为了提高山区的文化素质，政府又决定从m个村中选择n个村建小学（设 0 < n < = m < 500 ）。请根据给定的m、n以及所有相邻村庄的距离，选择在哪些村庄建小学，才使得所有村到最近小学的距离总和最小，计算最小值。

输入

第1行为m和n，其间用空格间隔
第2行为(m-1) 个整数，依次表示从一端到另一端的相邻村庄的距离，整数之间以空格间隔。

例如
10 3
2 4 6 5 2 4 3 1 3
表示在10个村庄建3所学校。第1个村庄与第2个村庄距离为2，第2个村庄与第3个村庄距离为4，第3个村庄与第4个村庄距离为6，...，第9个村庄到第10个村庄的距离为3。

输出

各村庄到最近学校的距离之和的最小值。

样例输入

10 2
3 1 3 1 1 1 1 1 3

样例输出

18

             传送门

             因为是一道动态规划题，我们定义f[i][j]=前i个村庄里修j个学校的最短距离和。结果则需输出f[m][n]。

             我们从学校数从小到大开始枚举，每次放入一个新学校时我们把每段决策分为两段：新学校位置前的村庄没放入新学校时的路程和剩下一段放入新学校(唯一的学校)的路程，累加后取最小值(注意初值)

             状态转移方程式：f[i][j]=min(f[i][j],f[k][j-1]+s[k+1][i]);

             具体见以下代码和每一个细节的注释

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<climits>
using namespace std;
int f[505][505],a[505],s[505][505],n,m;
//f[i][j]:前i个村建j个学校的路程最小值
//s[i][j]:从第i个村到第j个村修一个学校的最小值
//a[i]:第i个村离起点(第一个村)的距离
int mzls(int x,int y)//求s[i][j]的函数
{
	int ml=0,mid=(x+y)/2;//只有一个时肯定放中间
	for(int i=x;i<=y;i++)
		ml+=abs(a[i]-a[mid]);
	return ml;
}
int main()
{
	//freopen("T1.in", "r", stdin);
	//freopen("T1.out", "w", stdout);
	scanf("%d%d",&m,&n);
	for(int i=0;i<=m+1;i++)
		for(int j=0;j<=m+1;j++)
			f[i][j]=INT_MAX;
	for(int i=2;i<=m;i++)
	{
		scanf("%d",&a[i]);
		a[i]+=a[i-1];
	}
	for(int i=1;i<=m;i++)
		for(int j=i+1;j<=m;j++)
			s[i][j]=s[j][i]=mzls(i,j);
	for(int i=1;i<=m;i++)
		f[i][1]=s[1][i];//预处理只有一个的时候,避免第一次循环出错 
	for(int j=2;j<=n;j++)//放入第j个学校
		for(int i=j;i<=m;i++)//计算前i个村
			for(int k=j-1;k<i;k++)//枚举第j个学校放在哪里是最佳答案   
				f[i][j]=min(f[i][j],f[k][j-1]+s[k+1][i]);
	printf("%d\n",f[m][n]);
	return 0;
}