山区建小学(dp)

问题 J: 山区建小学

时间限制: 1 Sec   内存限制: 128 MB
提交: 15   解决: 3
[ 提交][ 状态][ 讨论版][命题人: quanxing]

题目描述

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

题解：
build[i][j]表示从i到j建一个学校,各村庄到学校的距离总和。
学校一定是建在i,j,中间。
len[i][j]表示有i个村庄一共建了j个学校的最优解。
则有len[i][j]=min(len[i][j],len[k][j-1]+build[k+1][i]);
代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=505;
int len[maxn][maxn],build[maxn][maxn],a[maxn];
int abs1(int x)
{
    if(x<0)return -x;
    return x;
}
int main()
{
    int n,m;scanf("%d%d",&m,&n);
    a[1]=0;
    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;j<=m;j++)
        {
            int mid=(i+j)>>1;build[i][j]=0;
            for(int k=i;k<=j;k++)
                build[i][j]+=abs1(a[mid]-a[k]);
        }
    }
    for(int i=0;i<505;i++)
        for(int j=0;j<505;j++)
        len[i][j]=1e9;
    for(int i=1;i<=m;i++)
    len[i][1]=build[1][i];
    for(int i=1;i<=m;i++)
    {
        for(int j=2;j<=n;j++)
        {
            for(int k=0;k<=i;k++)
                len[i][j]=min(len[i][j],len[k][j-1]+build[k+1][i]);
        }
    }
    printf("%d\n",len[m][n]);
    return 0;
}