① 动态规划② 线性动规 ③ 前缀和
一维和数组的b[i] 代表编号 1 到编号 i 的和,而二维和数组,变量是b[i][j] ,它代表左上角编号为 (1,1) ,右下角编号为 (i,j) 的矩阵的和。
for(i=1;i<=n;i++){ for(j=1;j<=m;j++){ b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+a[i][j];//核心代码 } }
//得到最大面的左上角坐标 For(i,1,n+1-c)For(j,1,m+1-c)//遍历二维前缀和数组 { if(s[i-1+c][j-1+c]-s[i-1+c][j-1]-s[i-1][j-1+c]+s[i-1][j-1]>maxnum)//注意减1容易打错 { maxnum=s[i+c-1][j+c-1]-s[i+c-1][j-1]-s[i-1][j+c-1]+s[i-1][j-1]; maxi=i; maxj=j; } } /*背诵方法 //得到最大面的左上角坐标 For(i,1,最大行数-矩形边长+1)For(j,1,最大列数-矩形边长+1)//遍历二维前缀和数组 { if(二维前缀和[当前行+矩形边长-1][当前列+矩形边长-1]-二维前缀和[当前行+矩形边长-1][当前列-1]-二维前缀和[当前行-1][当前列+矩形边长-1]+二维前缀和[当前行-1][当前列-1]>当前最大区域值) { 当前区域最大值=二维前缀和[当前行+矩形边长-1][当前列+矩形边长-1]-二维前缀和[当前行+矩形边长-1][当前列-1]-二维前缀和[当前行-1][当前列+矩形边长-1]+二维前缀和[当前行-1][当前列-1]; 最大区域左上角行下标=当前行; 最大区域左上角列下标=当前列; } } */
P2004 领地选择
for example:
- (1,1)(1,1) 为左上角,c=2 时我们算了(1,1)+(1,2)+(2,1)+(2,2);
- (1,2)(1,2) 为左上角,c=2 时我们算了(1,2)+(1,3)+(2,2)+(2,3);
咦?(1,2)+(2,2)被我们算重了!对于这种情况,有一个经典的解决方法——预处理。
所以我们考虑用(二维)前缀和优化
下面我们讲一下什么是二维前缀和,建立在一维前缀和之上,我们要求一个矩阵内一个任意的子矩阵的数的和,我们就可以用二维前缀和,我们还是用DP来预处理,状态和一维前缀和差不多,只不过我们多加了一维。
用 f(i,j) 表示 (1,1) 这个点与 (i,j) 这个点两个点分别为左上角和右下角所组成的矩阵内的数的和,则状态转移方程为:
f(i,j)=f(i,j−1)+f(i−1,j)−f(i−1,j−1)+vali,j
其中vali,j 表示 (i,j) 这一格的价值。
怎么来的呢?我们画一下图就知道了。
一开始是这样的:f(i,j) 为 0,即图中所有格子都是白色的。
回顾转移方程:f(i,j)=f(i,j−1)+f(i−1,j)−f(i−1,j−1)+vali,j
第一个加的是f(i,j−1),所以我们将 (1,1)(i,j−1) 染成黄色,表示加过一次。如上图所示。
此后再加上 f(i−1,j),但是我们发现两次染色中有一部分是重复染色的。即:(1,1)~(i−1,j−1),它们被染过两次色,在上图中我们将这一块区域染成棕色。
既然重复计算了,那我们就剪掉,即减去f(i−1,j−1),此时 (1,1)~(i−1,j−1) 又回归了黄色(即只染过一次色)。我们发现:只需再把(i,j) 这一格染成黄色,便大功告成了!
即加上:vali,j,如下图:
所以我们用图解证明了转移方程 f(i,j)=f(i,j−1)+f(i−1,j)−f(i−1,j−1)+vali,j
我们看一下实现代码:
for(int i=1; i<=m; i++)
for(int j=1; j<=n; j++)
f[i][j]=val[i][j]+f[i-1][j]+f[i][j-1]-f[i-1][j-1];
现在只需枚举左上角,根据边长 c 算出左下角,然后用二维前缀和 O(1) 更新。
附 (x1,y1) 到(x2,y2) 权值和的计算方法:
f(x2,y2)−f(x2,y1−1)−f(x1−1,y2)+f(x1−1,y1−1)
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cctype>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<vector>
#include<map>
#define For(i,a,b) for(i=(a);i<=(b);++i)
#define Forward(i,a,b) for(i=(a);i>=(b);--i)
using namespace std;
const int MAXN=21000;
template<typename T>
inline void read(T &x)//读入优化
{
T s=0,f=1;
char k=getchar();
while(!isdigit(k)&&k!='-')k=getchar();
if(k=='-')
{
f=-1;
k=getchar();
}
while(isdigit(k))
{
s=(s<<3)+(s<<1)+(k^'0');
k=getchar();
}
x=s*f;
}
int s[2000][2000],n,m,c;//s表示前i*j范围内所有点的权值和
int main(void)
{
read(n);
read(m);
read(c);
int i,j,k;
For(i,1,n)
For(j,1,m)
{
read(k);
s[i][j]=s[i-1][j]+s[i][j-1]+k-s[i-1][j-1];//预处理前缀和
}
int ma=s[c][c],ax=1,ay=1;
For(i,1,n-c+1)
For(j,1,m-c+1)
{
if(s[i+c-1][j+c-1]-s[i+c-1][j-1]-s[i-1][j+c-1]+s[i-1][j-1]>ma)//注意减1容易打错
{
ma=s[i+c-1][j+c-1]-s[i+c-1][j-1]-s[i-1][j+c-1]+s[i-1][j-1];
ax=i;
ay=j;
}
}
printf("%d %d\n",ax,ay);
return 0;
}