最大子矩阵题解

最大子矩阵

【题目描述】

已知矩阵的大小定义为矩阵中所有元素的和。给定一个矩阵，你的任务是找到最大的非空(大小至少是1×1)子矩阵。

比如，如下4×4的矩阵

0 -2 -7 0

9 2 -6 2

-4 1 -4 1

-1 8 0 -2

的最大子矩阵是

9 2

-4 1

-1 8

这个子矩阵的大小是15。

【输入】

输入是一个N×N的矩阵。输入的第一行给出N(0<N≤100)。再后面的若干行中，依次(首先从左到右给出第一行的N个整数，再从左到右给出第二行的N个整数……)给出矩阵中的N2个整数，整数之间由空白字符分隔(空格或者空行)。已知矩阵中整数的范围都在[−127,127]。

【输出】

输出最大子矩阵的大小。

【输入样例】

4 0 -2 -7 0 9 2 -6 2
-4 1 -4 1
-1 8 0 -2

【输出样例】

15

【来源】

一本通在线评测

题解

思路

首先枚举子矩阵的大小，然后算出子矩阵的和，最后将当前子矩阵的和与上一轮的相比较，留下最大的那个作答案，重复这个过程直到枚举完全部子矩阵

变量

int n,a[105][105],f[105][105],ans=-150,sum;

n记录大矩阵的大小；a记录矩阵中每个位置上的数字；f记录每个位置的前缀和（这个没看懂没关系，下面会详细讲)；ans记录答案；sum则用来临时储存当前子矩阵的和

前缀和

前缀和是一种快速计算子矩阵和的方法。具体步骤是先算出已1,1开始（也就是子矩阵左上角为1,1）的所有子矩阵的和，然后用几个的和相减，得出要求的子矩阵和

我们先来看如何计算已1,1开头的子矩阵的和

for(int i=1;i<=n;i++)//枚举终点横坐标
	{
		for(int j=1;j<=n;j++)//枚举终点纵坐标
		{
			for(int k=1;k<=i;k++)
			{
				for(int l=1;l<=j;l++)//遍历当前子矩阵中的每一个位置
				{
					f[i][j]+=a[k][l];//将这些位置上的数相加，最终得到当前子矩阵的和
				}
			}
		}
	}

找最大子矩阵

找最大子矩阵，首先要确定子矩阵的大小，这可以用两个参数确定：子矩阵的开头（左上角），和子矩阵的末尾（右下角）。确定这两个点，我们就知道子矩阵的长和宽了。

for(int i=1;i<=n;i++)//枚举子矩阵开头的横坐标
	{
		for(int j=1;j<=n;j++)//枚举子矩阵开头的纵坐标
		{
			for(int k=i;k<=n;k++)//枚举子矩阵末尾的横坐标
			{
				for(int l=j;l<=n;l++)//枚举子矩阵末尾的纵坐标
				{
					//用前缀和计算子矩阵的和
				}
			}
		}
	}

前缀和算法

如图所示，如果我们想要求蓝色子矩阵的和，那么就可以用有色区域的和减去灰色区域的和

而灰色区域又可以分为这么几部分：红色部分，黄色部分，和红黄都占有的橙色部分

那么当我们用有色部分前缀和（4,4的前缀和）减去灰色部分的前缀和时，实际上我们减的是红色部分（1,4）和黄色部分（4,1）再加上橙色部分（1,1）的前缀和，因为在减红色和黄色的时候多减去了一边橙色，所以要把它加回来一遍

这三个部分均可以依靠当前矩阵的位置求出。这里设i,j为子矩阵开头，k,l为子矩阵末尾，f记录前缀和。红色部分=f[k][j-1]，黄色部分=f[i-1][l]，橙色部分=f[i-1][j-1]

sum=f[k][l]-f[i-1][l]-f[k][j-1]+f[i-1][j-1];//当前子矩阵和=蓝-红-黄+橙
ans=max(ans,sum);//与上一次的子矩阵和进行比较，留下最大的作答案

输出

cout<<ans;

最后直接输出答案