快餐店

终于调出来了...

由于昨晚写了一半今天接着写的所以就没记时间/wq

分段 dp ,设 f[i][j] 表示前 i 个加油站中开设 j 个快餐店时最小的距离之和。

先预处理出在 i,j 之间只设一个快餐店时的代价(显然设在中间就好了)。

状态转移方程为 dp[i][j]=min(dp[i][j], dp[k][j-1]+cost[k+1][i]) 。

注意循环顺序,因为这个问题调了很久。

#include<bits/stdc++.h>
using namespace std;
int n,k,a[201],c[201][201],f[201][31];
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			f[i][j]=2e9;
			for(int k=i;k<=j;k++)
			{
				c[i][j]+=abs(a[(i+j)/2]-a[k]);
			}
		}
	}
	for(int i=1;i<=n;i++) f[i][1]=c[1][i];
	for(int j=2;j<=k;j++)
	{
		for(int i=j;i<=n;i++)
		{
			for(int l=j-1;l<=i-1;l++)
			{
				f[i][j]=min(f[i][j],f[l][j-1]+c[l+1][i]);
			}
		}
	}
	cout<<"Total distance sum = "<<f[n][k]<<endl;
	return 0;
}
posted @ 2020-12-12 23:12  樱雪喵  阅读(96)  评论(0编辑  收藏  举报