信息学奥赛一本通（1197：山区建小学）

1197：山区建小学

时间限制: 1000 ms         内存限制: 65536 KB
提交数: 3900     通过数: 2713

【题目描述】

        政府在某山区修建了一条道路，恰好穿越总共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

【分析】

        这道题是区间DP，放置到递推算法中有些牵强，详细分析可以参见：

https://www.cnblogs.com/TFLS-gzr/p/10809257.html。

        确定子问题：问题是m个村庄中选n个建小学，那么我们可以考虑在前m个村庄中选n-1个的情况，可以考虑在前m个村庄中选n-2个的情况，也可以考虑在前m-1个村庄中选n-1个的情况，……这些都是原问题的子问题。

　　设计状态：按照刚才所划分的子问题，我们可以自然的想到状态是什么，定义数组f[ i ][ j ]为当在前 i 个村庄中建立 j 个小学的最小路程和，那么原问题（在m个村庄中选n个建小学）的答案就是f[ m ][ n ]，这样问题就涉及好了。

阶段：村庄编号 m

状态：小学编号 n

决策：建？在哪建？

策略：最小路径和

目标：f[m][n]

　　状态转移方程：首先假装我们已经算出了每两个村庄之间建立一个小学的最短距离和，并把这些数据存在数组c[ i ][ j ]，那么我们就知道了村庄 i 到村庄 j 这个区间内建一个小学的最短距离和，先不要考虑我们是怎么算出来的；

边界：f[i][1]=c[1][i]，前 i 个村庄中建一个小学时，都等于1到第 i 个村庄只建一个学校的最短距离和。

【参考代码】

#include <stdio.h>
#include <math.h>
#define N 1010
int a[N][N]; //村庄i到村庄j间的最短距离
int c[N][N]; //村庄i到村庄j这个区间内建一个小学的最短距离和
int f[N][N]; //前i个村庄中建立j个小学的最小路程和
int min(int x,int y)
{
	return x < y ? x : y;
}
int main()
{
	int m,n,i,j,k,mid;
    scanf("%d%d",&m,&n);
    for(i=1;i<m;i++)
    	scanf("%d",&a[i][i+1]);

    for(i=1;i<=m;i++)
    	for(j=i+1;j<=m;j++)
    		a[i][j]=a[j][i]=a[i][j-1]+a[j-1][j];


    for(i=1;i<=m;i++)
    	for(j=i+1;j<=m;j++)
    	{
        	mid=(i+j)/2;
        	for(k=i;k<=j;k++)
        		c[i][j]+=a[k][mid];
    	}


    for(i=1;i<=m;i++)
    	f[i][1]=c[1][i];

    for(i=1;i<=m;i++)
    	for(j=2;j<=n;j++)
    	{
        	f[i][j]=999999;
        	for(k=j-1;k<=i;k++)
        	{
            	f[i][j]=min(f[i][j],f[k][j-1]+c[k+1][i]);
        	}
    	}

    printf("%d\n",f[m][n]);
    return 0;
}

http://ybt.ssoier.cn:8088/problem_show.php?pid=1197