描述
已知矩阵的大小定义为矩阵中所有元素的和。给定一个矩阵,你的任务是找到最大的非空(大小至少是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
解题思路
在只有一行的数组组成矩阵时我们可以轻松利用 dp[i] = max(dp[i-1] + a[i] , a[i] ) 找到以每个位置为终点的最大矩阵。
代码如下:
int findmax(int a[] , int n){
int temp[n+1] ;
temp[1] = a[1] ;
int m = temp[1] ;
for(int i = 2 ; i <= n ; i ++ ){
temp[i] = max(a[i] , a[i]+temp[i-1]) ;
if(temp[i] > m ) m = temp[i] ;
}
return m ;
}
而多行数据组成矩阵时,我们既然要让其成为最大矩阵,必然要将矩阵中的所有数字加起来,求和,然后找到各种情况的最大值。假设我们已知一个矩阵,我们可以先将每列的和算出,再将每列的和求和即是这个矩阵的所有元素的和。那我们不妨先将每一列的数值先加在一起,即可将原本二维的数组转化为一维数组求解。
至于这个矩阵的高度其实是不确定的,他的高度可以是1,2,3…n 。 每个情况都有可能。并且起始的行和终止的行也都不确定。这时候只能把所有的情况都列举出来。
在解题中定义一个tmp[N][N]数组。他的作用是将每一列的和记录下来。例如tmp[i][j]存放的是从 1----i 行所有j列元素的和。这样我们调用tmp[m][n] - tmp[j][n] 时可以得到j+1到m行所有j列元素的和。用两重循环就可以枚举所有的起始行和结束行情况。
对于每种情况求解,记录下最大值即是答案。
代码cpp
#include <bits/stdc++.h>
using namespace std;
const int N = 110 ;
int p[N][N] , tmp[N][N] , u[N]; //p用于读入数据,tmp作用如上文,u用来存储tmp相减后的值
int n , maxn = - 1 << 30 ;
int findmax(int a[] , int n){
int temp[n+1] ;
temp[1] = a[1] ;
int m = temp[1] ;
for(int i = 2 ; i <= n ; i ++ ){
temp[i] = max(a[i] , a[i]+temp[i-1]) ;
if(temp[i] > m ) m = temp[i] ;
}
return m ;
}
int main() {
cin >> n ;
for(int i = 1 ; i <= n ; i ++) {//输入
for(int j = 1 ; j <= n ; j++ ){
cin >> p[i][j] ;
}
}
//对tmp进行初始化,tmp第 0 行为0
for(int i = 1 ; i <= n ; i ++){
for(int j = 1 ; j <= n ; j++ ){
tmp[i][j] = tmp[i-1][j] + p[i][j] ;
}
}
//列举不同起始行和结束行
for(int i = 0 ; i < n ; i ++ ){
for(int j = i+1 ; j <= n ; j ++ ){
for(int k = 1 ; k <= n ; k++){
u[k] = tmp[j][k] - tmp[i][k] ;
}
maxn = max(maxn,findmax(u,n)) ;
}
}
cout << maxn ;
return 0;
}
OJ
http://noi.openjudge.cn/ch0206/1768/