最平坦的路线题解

题目描述

现有一个城市的地图，地图是一个$n\times m$的矩形，现在要在这个城市中旅行，旅行的起点左上角坐标为$(1,1)$，终点右下角坐标为$(n,m)$。地图上记录了每一个点的高度，坐标为$(x,y)$的位置，高度为$h_{x,y}$。每次只能向右或向下，不能走出城市，只能在整数点转弯。请你计算出一条路线，使得经过的点的高度的方差最小。输出最小的方差乘$(n+m-1{)}^2$后的值。

方差的定义：对于$k$个整数$x_1,x_2,\cdots ,x_k$，这$k$个数的方差为$S^2=\frac{(x_1-\overline{x}{)}^2+(x_2-\overline{x}{)}^2+\cdots +(x_k-\overline{x}{)}^2}{k}$，其中$\overline{x}$为这$k$个整数的平均值，即$\overline{x}=\frac{x_1+x_2+\cdots +x_k}{k}$。

输入格式

第一行输入$n,m$
下面$n$行，每行$m$个数，第$i$行第$j$列为$h_{i,j}$

输出格式

输出最小的方差乘$(n+m-1{)}^2$后的值

样例输入

2 2
1 2
3 4

样例输出

14

数据范围

对于$30\%$的数据，$1\le n,m\le 10$
对于$50\%$的数据，$1\le n,m\le 20$
对于$100\%$的数据，$1\le n,m\le 50,0\le h_{x,y}\le 50$

题解

可以看出要经过的点的数量为$n+m-1$，为方便表述，令$k=n+m-1$。

数据范围比较小，平均数又不好处理，我们可以考虑枚举平均数。因为$0\le \overline{x}\le 50$，所以$0\le k\cdot \overline{x}\le 50k$，且$k\cdot \overline{x}$为整数。于是我们就可以枚举$k\cdot \overline{x}$，再计算最大值。

将要求的式子拆一下，$k^2\times S^2=k^2\times \frac{(x_1-\overline{x}{)}^2+(x_2-\overline{x}{)}^2+\cdots +(x_k-\overline{x}{)}^2}{k}=k((\sum x_i^2)-2\overline{x}(\sum x_i)+k{\overline{x}}^2)$，在$\overline{x}$一定的情况下，$x_i$的贡献为$k(x_i^2-2\overline{x}{x}_i)=k{x}_i^2-2k\overline{x}{x}_i$。

设当前枚举的$k\cdot \overline{x}$值为$t$，设$f_{i,j}$表示从$(1,1)$到$(i,j)$的最小的贡献，则可以列出转移式
$$f_{i,j}=\min (f_{i-1,j},f_{i,j-1})+k\cdot h_{i,j}^2+2t\cdot h_{i,j}$$

那么答案为对于每一个$t$中$f_{n,m}+t^2$的最小值。

那如果这$k$个数的平均值乘$k$与$t$不相等怎么办，不就多算了几种情况吗？

会多一些情况，但不影响答案。设$t=k\overline{x}+a(a\ne 0)$，
此次计算的答案：$k(\sum x_i^2)-2t(\sum x_i)+t^2=k(\sum x_i^2)-2(k\overline{x}+a)(k\overline{x})+(k\overline{x}+a{)}^2$

实际上的答案：$k(\sum x_i^2)-2k\overline{x}(\sum x_i)+k^2{\overline{x}}^2=k(\sum x_i^2)-2(k\overline{x}{)}^2+k^2{\overline{x}}^2$

上下做差得$-2ak\overline{x}+2ak\overline{x}+a^2=a^2>0$，所以当$t\ne k\overline{x}$时，得出的答案会更大，而这个答案一定会被$t=k\overline{x}$时的答案覆盖，所以这是可行的。

枚举$t$是$O(50(n+m))$，每次DP都是$O(nm)$，所以总时间复杂度为$O(50nm(n+m))$。

code

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
long long ans=1e18,a[55][55],f[55][55];
int main()
{
	scanf("%d%d",&n,&m);k=n+m-1;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			scanf("%lld",&a[i][j]);
		}
	}
	for(int o=0;o<=5000;o++){
		f[1][1]=a[1][1]*a[1][1]*k-2*a[1][1]*o;
		for(int i=1;i<=n;i++){
			for(int j=1;j<=m;j++){
				if(i!=1||j!=1) f[i][j]=1e18;
				if(i>1) f[i][j]=min(f[i][j],f[i-1][j]+a[i][j]*a[i][j]*k-2*a[i][j]*o);
				if(j>1) f[i][j]=min(f[i][j],f[i][j-1]+a[i][j]*a[i][j]*k-2*a[i][j]*o);
			}
		}
		ans=min(ans,f[n][m]+o*o);
	}
	printf("%lld",ans);
	return 0;
}