山区建小学 动态规划

描述

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

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

解析：
dp[i][j]表示前i个村庄修j个小学的最小距离。
dp[i][j]=min(dp[i][j],dp[k][j-1]+d[k+1][i]); k在j-1到i之间
d[i][j]表示在i到j之间修一座小学的最小距离，该点在i到j中间

#include<stdio.h>
#include<math.h>
#define min(a,b) a<b?a:b

int dp[510][510],d[510][510],a[510];

int sum(int i,int j){//建一所学校的最小距离之和 
    int mid=(i+j)/2,l,ans=0;
    for(l=i;l<=j;l++){
        ans+=abs(a[l]-a[mid]);
    }
    return ans;
}

int main(){
    int i,j,k,n,m;
    scanf("%d%d",&m,&n);
    for(i=2;i<=m;i++){
        scanf("%d",&a[i]);
        a[i]+=a[i-1];
    }
    //i到j之间建一所学校的最小距离 
    for(i=1;i<=m;i++){
        for(j=i;j<=m;j++){
            d[i][j]=sum(i,j);
        }
    }
    //将已知的dp求出 
    dp[0][0]=0;
    for(i=1;i<=m;i++){
        dp[i][1]=d[1][i];
    }
    //动态规划 
    for(i=n;i<=m;i++){
        for(j=2;j<=n&&j<=i;j++){//j=1时dp[i][j]=d[1][i];
            dp[i][j]=dp[i-1][j-1];
            for(k=i-2;k>=j-1;k--){
                dp[i][j]=min(dp[i][j],dp[k][j-1]+d[k+1][i]);
            }
        }
    }
    printf("%d",dp[m][n]);
}