hduoj1559 最大子矩阵-二维前缀和-CSDN博客

本文链接：https://blog.csdn.net/m0_74157667/article/details/129803615

该问题要求找到一个m×n矩阵中，x×y的子矩阵，其元素和最大。使用二维前缀和技巧，可以将时间复杂度降低到O(n^4)，通过计算每个位置的子矩阵和来找到最大值。

摘要由CSDN通过智能技术生成

给你一个m×n的整数矩阵，在上面找一个x×y的子矩阵，使子矩阵中所有元素的和最大。

Input

输入数据的第一行为一个正整数T，表示有T组测试数据。每一组测试数据的第一行为四个正整数m,n,x,y（0<m,n<1000 AND 0<x<=m AND 0<y<=n），表示给定的矩形有m行n列。接下来这个矩阵，有m行，每行有n个不大于1000的正整数。

Output

对于每组数据，输出一个整数，表示子矩阵的最大和。

Sample Input

1
4 5 2 2
3 361 649 676 588
992 762 156 993 169
662 34 638 89 543
525 165 254 809 280

Sample Output

2474

给定一个n*n的矩阵，找一个最大的子矩阵，使得这个子矩阵里面的元素和最大。

这道题最朴素的算法是 O(n ^ 6),用二维前缀和可以降到 O(n ^ 4)。

再想想前缀和矩阵怎么生成？看个图就明白了：

那么最大的矩形前缀和就等于蓝的矩阵加上绿的矩阵，再减去重叠面积，最后加上小方块，即

sum[i][j] = sum[i][j - 1] + sum[i - 1][j] - sum[i - 1][j - 1] + a[i][j]

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 1001;
typedef long long LL;
int a[N][N];
LL sum[N][N];
int main()
{
	int t;
	
	scanf("%d",&t);
	while(t--)
	{int m,n,x,y;
		memset(sum,0,sizeof(sum));
		scanf("%d %d %d %d",&m,&n,&x,&y);
		for(int i=1;i<=m;i++)
		{
			for(int j=1;j<=n;j++)
			{
				scanf("%d",&a[i][j]);
				sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
			}//二维前缀和
		}
		LL ans=0;
		for(int i=x;i<=m;i++)
		{
			for(int j=y;j<=n;j++)
			{
				ans=max(ans,sum[i][j]-sum[i-x][j]-sum[i][j-y]+sum[i-x][j-y]);
			}
		}
		printf("%lld\n",ans);
	}
}