洛谷 P4677 山区建小学【dp】

---

题目：

传送门

---

题意：

在$n$村庄中选出$m$个，使得所有村庄到达选定村庄的总路程最短

---

分析：

设$an{s}_{i,j}$表示在前$i$个村庄中选出$j$个时的答案，$f_{i,j}$表示在$i$到$j$这几个村庄只选$1$时的最优解
我们用$k$来枚举倒数第二个选定的村庄是哪一个，就有方程
a n s i , j = m i n { a n s k , j − 1 + f k + 1 , i } ans_{i,j}=min\{ans_{k,j-1}+f_{k+1,i}\} ansi,j​=min{ansk,j−1​+fk+1,i​}
现在我们只需要求出$f$就行了
我们根据多年义务教育积累下来的经验，可以得到选择中点必定是最优的选择

---

代码：

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
#include<vector>
#define LZX Mu
#define LL long long 
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}
int dis[505],f[505][505],sum[505],ans[505][505];
int main()
{
	int n=read(),m=read();
	for(int i=1;i<n;i++) dis[i]=read(),sum[i+1]=sum[i]+dis[i];
	for(int i=1;i<=n;i++)
	  for(int j=1;j<=n;j++)
	  	for(int k=i;k<=j;k++) f[i][j]+=abs(sum[(i+j)>>1]-sum[k]);
	memset(ans,0x3f,sizeof(ans));
	ans[0][0]=0;
	for(int i=1;i<=n;i++)
	  for(int j=1;j<=m;j++)
	  {
	  	if(j>i) {ans[i][j]=0;continue;}
	  	for(int k=j-1;k<=i;k++) ans[i][j]=min(ans[i][j],ans[k][j-1]+f[k+1][i]);
	  }
	cout<<ans[n][m];
	return 0;
}