【DP】 noi openjudge 2.6 山区建小学

7624:山区建小学

总时间限制: 

1000ms 

内存限制: 

65536kB

描述

政府在某山区修建了一条道路，恰好穿越总共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[n][k]为在前n个村庄建立k个小学的最小距离

	g[i][j]表示在第i个村与第j个村之间建立一个小学的最小距离

	那么不难得到f[n][k]=min{f[n-1][k-1],f[n-2][k-1]+g[n-1][n]...f[k-1][k-1]+g[k][n]};

     再来求g的值，把小学在i与j之间的哪个位置？若设建在x点，则g[i][j]=

	|x-i|+|x-(i+1)|+|x-(i+2)|....+|x-j|

        由绝对值的几何意义（不懂的翻翻初一的数学笔记吧）可得x建在i,j中点距离最小，g[i][j]的值就确定了

     现在轮到赋初值了，因为最后求f[n][k],所以循环为顺序，只需给f[i][1]赋值就可以了

	在前i个村庄建立1个小学，不就是在第1个村与第i个村之间建一个小学吗？

	所以f[i][1]=g[1][i];

代码如下：

#include<cstdio>
#include<algorithm>
using namespace std;
int f[510][510],g[510][510],a[510];
int xmy(int l,int q,int w)
{
	int s=0;
	for(int i=q;i<=w;i++)
		s+=abs(a[l]-a[i]);
	return s;
}
int main()
{
	int n,k,i,j,u,l;
	scanf("%d%d",&n,&k);
	for(i=2;i<=n;a[i]+=a[i-1],i++)
		scanf("%d",&a[i]);
	for(i=1;i<=n;i++)
		for(j=i+1;j<=n;j++)
		{
			l=(i+j)/2;
			g[i][j]=xmy(l,i,j);
			if(i==1) f[j][1]=g[1][j];
		}
	for(i=3;i<=n;i++)
		for(j=2;j<i&&j<=k;j++)
		{
			f[i][j]=f[i-1][j-1];
			for(u=i-2;u>=j-1;u--)
				f[i][j]=min(f[i][j],f[u][j-1]+g[u+1][i]);
		}
	printf("%d",f[n][k]);
}