洛谷P4677 山区建小学

题目描述

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

输入格式

第一行为 n 和 m，其间用空格间隔。

第二行为 n−1 个整数，依次表示从一端到另一端的相邻村庄的距离，整数之间以空格间隔。

例如

10 3
2 4 6 5 2 4 3 1 3

表示在 1010 个村庄中建 33 所学校。第 11 个村庄与第 22 个村庄距离为 22，第 22 个村庄与第 33 个村庄距离为 44，第 33 个村庄与第 44 个村庄距离为 66，...，第 99 个村庄到第 1010 个村庄的距离为 33。

输出格式

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

输入输出样例

输入 #1复制

10 2
3 1 3 1 1 1 1 1 3

输出 #1复制

18

这是一道典型的递推。

AC代码

#include<bits/stdc++.h>
using namespace std;
 
const int N=510;
 
int n,m;
int d[N],s[N];
int f[N][N],dp[N][N];
 
int main(){
    cin>>n>>m;
    for(int i=1;i<=n-1;i++){
        cin>>d[i];
        s[i+1]=s[i]+d[i];
    }
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j++){
            int mid=i+j>>1;
            for(int k=i;k<=mid-1;k++){
                f[i][j]=f[i][j]+s[mid]-s[k];
            }
            for(int k=mid+1;k<=j;k++){
                f[i][j]=f[i][j]+s[k]-s[mid];
            }
        }
    }
    for(int i=1;i<=n;i++){
        dp[i][1]=f[1][i];
    }
    for(int i=1;i<=n;i++){
        for(int j=2;j<=m&&j<i;j++){
            dp[i][j]=1e8;
            for(int k=j-1;k<=i-1;k++){
                dp[i][j]=min(dp[i][j],dp[k][j-1]+f[k+1][i]);
            }
        }
    }
    cout<<dp[n][m]<<endl;
    return 0;
}

求赞！！！