题解 P4677 【山区建小学】

我用的是递推的思路来做(看注释)

#include<bits/stdc++.h>//递推
using namespace std;
const int M=501;
int a[M][M],c[M][M],f[M][M];
int main()
{
    int n,t,m,mid;
    cin>>m>>n;//按题意先输入n个村庄和m个小学
    for(int i=1;i<m;i++)//输入
        cin>>a[i][i+1];
    for(int i=1;i<=m;i++)
        for(int j=i+1;j<=m;j++){
            a[i][j]=a[j-1][j]+a[i][j-1];//初值
            a[j][i]=a[i][j];//a数组表示ij两地之间的距离
        }
    for(int i=1;i<=m;i++)
        for(int j=i+1;j<=m;j++){
            mid=(i+j)/2;//在山地中间建小学最优
            c[i][j]=0;//此时现将c[i][j]变成0
            for(int k=i;k<=j;k++)
               c[i][j]+=a[k][mid];//c数组统计ij两地到学校的距离
        }
    for(int i=1;i<=m;i++)
        f[i][1]=c[1][i];//将f数组设初值
    for(int i=1;i<=m;i++)
        for(int j=2;j<=n;j++){
            f[i][j]=214748364;//先将f[i][j]设为最大值
            for(int k=j-1;k<=i;k++)
                f[i][j]=min(f[i][j],f[k][j-1]+c[k+1][i]);//找出学校位置的最小值并替换
        }
    cout<<f[m][n];//输出最后答案
    return 0;
}