1768:最大子矩阵

描述

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

 

思路

这个题目是字段和的二维情况，处理起来稍微复杂一点，一开始也想了好久，才想明白。

先求每行元素的前n项和存储到对应位置，想要取某几列元素和时相减就可以得到，例如求3~5列元素和，只需要用num[5]-num[3-1]就可以得到这行的3~5列元素和。

然后对列依次相减取不同列的组合，在行上就类似于一维字段和的处理求出最大和就好了。

代码

#include <iostream>
using namespace std;
int num[105][105];
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		int sum=0;
		for(int j=1;j<=n;j++){
			int a;
			cin>>a;
			sum+=a;
			num[i][j]=sum; 
		}
	}

	int ans=num[1][1];
	for(int j=1;j<=n;j++){
		for(int k=j;k<=n;k++){
			int count=0;
			for(int i=1;i<=n;i++){
				count+=(num[i][k]-num[i][j-1]);
				if(count>ans) ans=count;
				if(count<0) count=0;
			}
		}
	}

	cout << ans <<endl;
	return 0;
}

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