IOI 2000 邮局 题解(1)

题目传送门

题目大意： 数轴上有 $n$ 个点，给出他们的坐标 $s_i$，现在要选一些点建邮局，每个点的代价是离他最近的邮局的距离，找到一种方案使所有点总代价最小。

题解

先考虑 $40$ 分的暴力 $O(n^3)$ $dp$，设 $f[i][j]$ 表示前 $i$ 个点放 $j$ 个邮局的最小代价，那么有：
f [ i ] [ j ] = min ⁡ k = 1 j { f [ k ] [ j − 1 ] + w ( k + 1 , i ) } f[i][j]=\min_{k=1}^j\{f[k][j-1]+w(k+1,i)\} f[i][j]=k=1minj​{f[k][j−1]+w(k+1,i)}

其中，$w(k+1,i)$ 表示在 $k+1$ 到 $i$ 之间放一个邮局的最小代价，显然，$w$ 满足 $w(i,i)=0$。

显然求 $w(a,b)$ 就是找到 $[a,b]$ 这一段的中位数（如果有偶数个数，那么中间的两个数其实都是可以选的），然后求出这一段的所有点到中位数的距离之和，这个预处理一下前缀和就好。（如果不明白建议恶补入门贪心qwq）

诶，这个方程长的眉清目秀，莫不是可以用四边形不等式来优化一波？

接下来就是走流程了，一步一步证明下去即可。（如果发现下面有什么跃迁式证明，你可能是没看上面那个链接，不过学过四边形不等式的应该都看得懂qwq）

first

首先证明 $w$ 满足四边形不等式，根据推论，也就是要证明：对于任意 $a<c$，都有 $w(a,c)+w(a+1,c+1)\le w(a+1,c)+w(a,c+1)$。

证明：

因为 $w(a+1,c+1)$ 和 $w(a,c+1)$ 可以取到同样的中位数 $mi{d}_1$，所以 $w(a,c+1)$ 比 $w(a+1,c+1)$ 大 $(s_{mi{d}_1}-s_a)$。

同样的，$w(a,c)$ 比 $w(a+1,c)$ 大 $(s_{mi{d}_2}-s_a)$。

于是就有：
$$\begin{gathered} w(a,c)+w(a+1,c+1)-(w(a+1,c)+w(a,c+1))= \\ {s}_{mi{d}_2}-s_a-(s_{mi{d}_1}-s_a)=s_{mi{d}_2}-s_{mi{d}_1} \end{gathered}$$

由于 $mi{d}_2<mi{d}_1$，所以 $s_{mi{d}_2}\le s_{mi{d}_1}$，即 $s_{mi{d}_2}-s_{mi{d}_1}\le 0$，于是得到：
$$\begin{gathered} w(a,c)+w(a+1,c+1)-(w(a+1,c)+w(a,c+1))\le 0 \\ w(a,c)+w(a+1,c+1)\le w(a+1,c)+w(a,c+1) \end{gathered}$$

证毕。

second

接下来要证明的是对于任意 $a\le b\le c\le d$，都有 $w(a,d)\ge w(b,c)$。

$a$ 到 $d$ 的数字肯定比 $b$ 到 $c$ 数字多，那么到中位数的距离之和显然会变大……

third

当我写完的时候我发现自己搞错了一些定义，只好从头再来……

接下来证明 $f$ 满足四边形不等式，即证明对于任意 $a<c$，都有 $f[a][c]+f[a+1][c+1]\le f[a+1][c]+f[a][c+1]$。

先考虑 $c=1$ 的情况，此时有：
$$\begin{gathered} f[a][c]+f[a+1][c+1]=f[a][1]+f[a+1][2]=f[a+1][2]+w(1,a) \\ f[a+1][c]+f[a][c+1]=f[a+1][1]+f[a][2]=f[a][2]+w(1,a+1) \end{gathered}$$

相减得到：
$$f[a+1][2]-f[a][2]+w(1,a)-w(1,a+1)$$

设 $f[a][2]$ 的最优决策为 $p$，那么有 $f[a][2]=w(1,p)+w(p+1,a)$。

这个决策对于 $f[a+1][2]$ 来说不一定最优，所以 $f[a+1][2]\le w(1,p)+w(p+1,a+1)$

将这两个柿子代入上面的柿子得到：
$$\begin{gathered} f[a+1][2]-f[a][2]+w(1,a)-w(1,a+1)\le \\ w(1,p)+w(p+1,a+1)-w(1,p)-w(p+1,a)+w(1,a)-w(1,a+1)= \\ w(p+1,a+1)-w(p+1,a)+w(1,a)-w(1,a+1) \end{gathered}$$

因为 $w$ 满足四边形不等式，所以有：
$$\begin{gathered} w(1,a)+w(p+1,a+1)\le w(1,a+1)+w(p+1,a) \\ w(1,a)+w(p+1,a+1)-w(1,a+1)-w(p+1,a)\le 0 \end{gathered}$$

代入上面的柿子，最终得到：
$$\begin{gathered} f[a+1][2]-f[a][2]+w(1,a)-w(1,a+1)\le 0 \\ f[a+1][2]+w(1,a)\le f[a][2]+w(1,a+1) \end{gathered}$$

即 $f[a][c]+f[a+1][c+1]\le f[a+1][c]+f[a][c+1]$。

接下来考虑 $\forall c<a$ 的情况，假设此时 $c=k$，以及 $c<k$ 的情况已经证明完了。

设 $f[a+1][c]$ 和 $f[a][c+1]$ 的最优决策点分别为 $x,y$，那么有：
$$\begin{gathered} f[a+1][c]=f[x][c-1]+w(x+1,a+1) \\ f[a][c+1]=f[y][c]+w(y+1,a) \end{gathered}$$

这两个决策点对于 $f[a][c]$ 和 $f[a+1][c+1]$ 不一定最优，下面分类讨论：

当 $x\le y$ 时，有：
$$\begin{gathered} f[a][c]\le f[x][c-1]+w(x+1,a) \\ f[a+1][c+1]\le f[y][c]+w(y+1,a+1) \end{gathered}$$

综合一下上面的柿子，可以得到：
$$\begin{gathered} f[a+1][c]+f[a][c+1]-f[a][c]-f[a+1][c+1](1)\ge \\ w(x+1,a+1)+w(y+1,a)-w(x+1,a)-w(y+1,a+1) \end{gathered}$$

因为 $w$ 满足四边形不等式，所以有：
$$w(x+1,a+1)+w(y+1,a)\ge w(x+1,a)+w(y+1,a+1)$$

于是得到：$(1)\ge 0$，即 $f[a][c]+f[a+1][c+1]\le f[a+1][c]+f[a][c+1]$。

当 $x>y$ 时，$x$ 可能等于 $a$，而 $a$ 不能是 $f[a][c]$ 的决策点，所以这里将 $x$ 代入给 $f[a+1][c+1]$：
$$\begin{gathered} f[a][c]\le f[y][c-1]+w(y+1,a) \\ f[a+1][c+1]\le f[x][c]+w(x+1,a+1) \end{gathered}$$

综合一下上面的柿子，可以得到：
$$\begin{gathered} f[a+1][c]+f[a][c+1]-f[a][c]-f[a+1][c+1](2)\ge \\ f[x][c-1]+f[y][c]-f[y][c-1]-f[x][c] \end{gathered}$$

根据假设，可以得到：
$$\begin{gathered} f[x][c-1]+f[y][c]\ge f[y][c-1]+f[x][c] \\ f[x][c-1]+f[y][c]-f[y][c-1]-f[x][c]\ge 0 \end{gathered}$$

于是得到：$(2)\ge 0$，即 $f[a][c]+f[a+1][c+1]\le f[a+1][c]+f[a][c+1]$。

forth

接下来要证明的就是最优决策点的优化了：$p(i,j-1)\le p(i,j)\le p(i+1,j)$。

设 $p=p(i,j)$，$k$ 满足 $i\le k<p$。

假设 $f[i+1][j]$ 以 $p,k$ 作为决策点，那么就有：
$$\begin{gathered} k:f[i+1][j]=f[k][j-1]+w(k+1,i+1) \\ p:f[i+1][j]=f[p][j-1]+w(p+1,i+1) \end{gathered}$$

两式相减得到：
$$f[k][j-1]-f[p][j-1]+w(k+1,i+1)-w(p+1,i+1)(1)$$

假设 $f[i][j]$ 以 $p,k$ 作为决策点，则有：
$$\begin{gathered} k:f[i][j]=f[k][j-1]+w(k+1,i) \\ p:f[i][j]=f[p][j-1]+w(p+1,i) \end{gathered}$$

由于 $p$ 是 $f[i][j]$ 的最优决策点，所以两式相减得到：
$$f[k][j-1]-f[p][j-1]+w(k+1,i)-w(p+1,i)\ge 0(2)$$

因为 $w$ 满足四边形不等式，所以有：
$$\begin{gathered} w(p+1,i+1)+w(k+1,i)\le w(k+1,i+1)+w(p+1,i) \\ w(k+1,i+1)-w(p+1,i+1)\ge w(k+1,i)-w(p+1,i) \end{gathered}$$

将这个柿子和 $(2)$ 式代入 $(1)$ 式，得到：
$$\begin{gathered} f[k][j-1]-f[p][j-1]+w(k+1,i+1)-w(p+1,i+1)\ge \\ f[k][j-1]-f[p][j-1]+w(k+1,i+1)-w(p+1,i+1)\ge 0 \end{gathered}$$

即 $(1)\ge 0$，所以对于 $f[i+1][j]$，$p$ 一定是比 $k$ 优的，所以它的决策点一定不可能小于 $p$，即 $p(i,j)\le p(i+1,j)$。

同理，可以得到 $p(i,j-1)\le p(i,j)$。

last

然后用最后的结论优化 $dp$ 就好了。

代码如下：

#include <cstdio>
#include <cstring>
#define maxn 3010

int n,m,a[maxn];
int sum[maxn],f[maxn][310],p[maxn][maxn];
int w(int x,int y)
{
	int mid=x+y>>1;
	return sum[y]-sum[mid]-(y-mid)*a[mid]+(mid-x)*a[mid]-(sum[mid-1]-sum[x-1]);
}

int main()
{
	scanf("%d %d",&n,&m);
	memset(f,63,sizeof(f));
	for(int i=1;i<=n;i++)scanf("%d",&a[i]),
	sum[i]=sum[i-1]+a[i],f[i][1]=w(1,i),p[i][1]=1;
	for(int j=2;j<=m;j++)
	{
		p[n+1][j]=n;
		for(int i=n;i>=1;i--)
		for(int k=p[i][j-1];k<=p[i+1][j];k++)
		if(f[k][j-1]+w(k+1,i)<f[i][j])f[i][j]=f[k][j-1]+w(k+1,i),p[i][j]=k;
	}
	printf("%d",f[n][m]);
}

这题的另外一个做法在这里。