【动态规划】求最大加权矩形的两种方法 二维前缀和 矩阵压缩

矩阵压缩

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			scanf("%d",&x);
			a[i][j]=a[i-1][j]+x; //求列前缀和
		}
	}
	for(int i=1;i<=n;i++){
		for(int k=1;k<=i;k++){
			int f[N]={0},dp[N]={0};
			for(int j=1;j<=n;j++){
				f[j]=a[i][j]-a[i-k][j]; //即为从第i-k+1行到第i行的矩阵 列向压缩成一维f[]
				dp[j]=max(dp[j-1]+f[j],f[j]); //将一列数当成一个数进行dp
				ans=max(dp[j],ans);
			}
		}
	}
	printf("%d\n",ans);
} 

二维前缀和

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			scanf("%d",&x);
			a[i][j]=a[i][j-1]+x; //横向压缩 a[i][j]为从a[i][1]到a[i][j]的行前缀和
			sum[i][j]=sum[i-1][j]+a[i][j]; //两个前缀和之和
		}
	}
	for(int x1=1;x1<=n;x1++){
		for(int x2=1;x2<=n;x2++){
			for(int y3=1;y3<=n;y3++){
				for(int y4=1;y4<=n;y4++){
					if(x2<x1||y4<y3) continue;
					ans=max(ans,sum[x2][y4]+sum[x1-1][y3-1]-sum[x1-1][y4]-sum[x2][y3-1]);
				}
			}
		}
	}
	printf("%d\n",ans);
}