最大子矩阵题解

最大子矩阵题解

题目描述

已知矩阵的大小定义为矩阵中所有元素的和。给定一个矩阵，你的任务是找到最大的非空(大小至少是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

输入是一个N×N的矩阵。输入的第一行给出N

输入格式

第一行为 $n,m,k$，接下来 $n$ 行描述矩阵每行中的每个元素的分值（每个元素的分值的绝对值不超过 $32767$）。

输出格式

输出最大子矩阵的大小。

样例 #1

样例输入 #1

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

样例输出 #1

15

思路1

使用枚举的方法，需要使用六层循环，只能拿70分

1.输入

建立$a$数组，输入。

#include<iostream>
#include<cmath>
#include<iomanip>
#include<cstring>
#include<stack>
#include<queue>
#include<algorithm>
using namespace std;
int a[105][105],f[101][101];
int main()
{
	int n,ix,iy,jx,jy,i,j,k,sum,maxx=-0x3f3f3f;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=n;j++) cin>>a[i][j];
	}
	return 0;
}

2.枚举

使用六层循环，前四层分别枚举子矩阵左上角、右下角的x、y坐标，五、六层循环遍历子矩阵，求和并寻找最大值。
注意：最大值变量$maxx$的初始值。

#include<iostream>
#include<cmath>
#include<iomanip>
#include<cstring>
#include<stack>
#include<queue>
#include<algorithm>
using namespace std;
int a[105][105],f[101][101];
int main()
{
	int n,ix,iy,jx,jy,i,j,k,sum,maxx=-0x3f3f3f;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=n;j++) cin>>a[i][j];
	}
	for(ix=1;ix<=n;ix++)
	{
		for(iy=1;iy<=n;iy++)
		{
			for(jx=ix;jx<=n;jx++)
			{
				for(jy=iy;jy<=n;jy++)
				{
					sum=0;
					for(i=ix;i<=jx;i++)
					{
						for(j=iy;j<=jy;j++) sum+=a[i][j];
					}
					maxx=max(maxx,sum);
				}
			}
		}
	}
	cout<<maxx;
	return 0;
}

思路2

用每一个元素的前缀和求最大值，是满分算法。
输入和思路1的输入是一样的。

1.前缀和

前缀和指的是从以每一个元素为小矩阵的右下角，以（1，1）的元素为头的矩阵元素之和。
所以我们写一段循环求出矩阵每一个元素的前缀和,存入$f$数组中。
循环使用四层循环，两层由于遍历元素的坐标，两层用于遍历子矩阵求前缀和，这是不会超时的，$N$最大是100，也就是最多运算10^8次。

#include<iostream>
#include<cmath>
#include<iomanip>
#include<cstring>
#include<stack>
#include<queue>
#include<algorithm>
using namespace std;
int a[105][105],f[101][101];
int main()
{
	int n,ix,iy,jx,jy,i,j;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=n;j++) cin>>a[i][j];
	}
	for(ix=1;ix<=n;ix++)
	{
		for(iy=1;iy<=n;iy++)
		{
			for(i=1;i<=ix;i++)
			{
				for(j=1;j<=iy;j++) f[ix][iy]+=a[i][j];
			}
		}
	}
	return 0;
}

2.求每一个子矩阵的和以及其最大值

对于任意一个子矩阵，我们在知道大矩阵前缀和的情况下，可以用如下方法求和：

图中每一个框是一个子矩阵，用$S$记录每个子矩阵右下角元素的前缀和。
黑色框是大矩阵，红色框是要求和的矩阵，记作$S_0$.
橙色框记作$S_1$蓝色框记作$S_2$,绿色框记作$S_3$,粉色框记作$S_3$.
由图，我们可以得到一个公式：
$S_0=S_1-S_2-S_3+S_4$
我们已经掌握了求任意一个子矩阵的方法，我们就可以枚举子矩阵的行，列坐标，求出每个子矩阵的和，进而得到答案。

#include<iostream>
using namespace std;
int a[105][105],f[101][101];
int main()
{
	int n,ix,iy,jx,jy,i,j,sum,maxx=-0x3f3f3f;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=n;j++) cin>>a[i][j];
	}
	for(ix=1;ix<=n;ix++)
	{
		for(iy=1;iy<=n;iy++)
		{
			for(i=1;i<=ix;i++)
			{
				for(j=1;j<=iy;j++) f[ix][iy]+=a[i][j];
			}
		}
	}
	for(ix=1;ix<=n;ix++)//枚举左上角行坐标
	{
		for(iy=1;iy<=n;iy++)//枚举左上角列坐标
		{
			for(jx=ix;jx<=n;jx++)//枚举右下角行坐标
			{
				for(jy=iy;jy<=n;jy++)//枚举右下角列坐标
				{
					sum=f[jx][jy]-f[ix-1][jy]-f[jx][iy-1]+f[ix-1][iy-1];//求子矩阵的和
					maxx=max(maxx,sum);//求最大值
				}
			}
		}
	}
	cout<<maxx;
	return 0;
}

AC代码

#include<iostream>
#include<cmath>
#include<iomanip>
using namespace std;
int a[105][105],f[101][101];
int main()
{
	int n,ix,iy,jx,jy,i,j,maxx=-0x3f3f3f;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=n;j++) cin>>a[i][j];
	}
	for(ix=1;ix<=n;ix++)
	{
		for(iy=1;iy<=n;iy++)
		{
			for(i=1;i<=ix;i++)
			{
				for(j=1;j<=iy;j++) f[ix][iy]+=a[i][j];
			}
		}
	}
	for(ix=1;ix<=n;ix++)
	{
		for(iy=1;iy<=n;iy++)
		{
			for(jx=ix;jx<=n;jx++)
			{
				for(jy=iy;jy<=n;jy++)
				{
					sum=f[jx][jy]-f[ix-1][jy]-f[jx][iy-1]+f[ix-1][iy-1];
					maxx=max(maxx,sum);
				}
			}
		}
	}
	cout<<maxx;
	return 0;
}

题目来源

最大子矩阵
最大子矩阵