P4767 [IOI2000]邮局（四边形不等式）

Problem

高速公路旁边有一些村庄。高速公路表示为整数轴，每个村庄的位置用单个整数坐标标识。没有两个在同样地方的村庄。两个位置之间的距离是其整数坐标差的绝对值。

邮局将建在一些，但不一定是所有的村庄中。为了建立邮局，应选择他们建造的位置，使每个村庄与其最近的邮局之间的距离总和最小。

你要编写一个程序，已知村庄的位置和邮局的数量，计算每个村庄和最近的邮局之间所有距离的最小可能的总和。村庄总数<=3000，邮局总数<=300。

Solution

预处理出w(i,j)表示i~j个村庄放一个邮局的最小距离和，放的位置为中位数，可递推。考虑DP，dp(i,j)表示前i个村庄放j个邮局的最小距离和，dp(i,j)=min{dp(k,j-1)+w(k+1,i)}。由于w(i,j)满足四边形不等式(w(a,b)+w(c,d)<=w(a,d)+w(b,c),a<b<=c<d,即交叉小于包含)，dp同样满足上述关系，s(i,j)表示得到dp(i,j)的k的位置，s(i,j)<=s(i,j+1)<=s(i+1,j+1)，因此可维护转移的位置来减少总转移量，时间复杂度为O(PV)。

Code

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
int n,m,pos[3005],w[3005][3005],dp[3005][305],s[3005][3005];
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	scanf("%d",&pos[i]);
	sort(pos+1,pos+1+n);
	for(int i=1;i<=n;i++)
	for(int j=i+1;j<=n;j++)
	w[i][j]=w[i][j-1]+pos[j]-pos[(i+j)/2];
	memset(dp,0x3f,sizeof(dp));
	dp[0][0] = 0;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			for(int k=s[i][j-1];k<i;k++)
			{
				if(dp[i][j]>dp[k][j-1]+w[k+1][i])
				{
					dp[i][j] = dp[k][j-1]+w[k+1][i];
					s[i][j] = k;
				} 
			}
		}
	}
	cout<<dp[n][m];
	return 0;
}