题目传送门
题目描述
已知矩阵的大小定义为矩阵中所有元素的和。给定一个矩阵,你的任务是找到最大的非空(大小至少是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
思路
转化为最大子序列和计算,枚举遍历起始行 i 和终止行 j, i = j i=j i=j,寻找原矩阵的该行的最大子序列和, i ≠ j i≠j i=j 的话 将 i , i+1 , … , j 的各行元素加到一行上,然后计算这一行的最大子序列和。
构造一个矩阵total,便于计算每行的叠加和, t o t a l [ i ] [ j ] = ∑ x = 0 i t o t a l [ x ] [ j ] total[i][j]=\sum_{x=0}^{i} total[x][j] total[i][j]=∑x=0itotal[x][j]
help为转化的一维数组
代码
#include <bits/stdc++.h>
#define MAXN 105
using namespace std;
int arr[MAXN][MAXN];
int total[MAXN][MAXN];
int help[MAXN];
int dp[MAXN];
int n;
void init(){
memset(arr,0,MAXN);
memset(total,0,MAXN);
memset(help,0,MAXN);
memset(dp,0,MAXN);
}
int findMax(){
dp[0] = help[0];
int ans = dp[0];
for (int i=1;i<n;i++){
dp[i]=max(help[i],dp[i-1]+help[i]);
ans=max(dp[i],ans);
}
return ans;
}
int getArr(){
int ans = -1;
for (int i=0;i<n;i++){
for (int j=i;j<n;j++){
for (int k=0;k<n;k++){
if (i==0){
help[k]=total[j][k];
}else {
help[k]=total[j][k]-total[i-1][k];
}
}
ans = max(ans,findMax());
}
}
return ans;
}
int main (){
while (scanf("%d",&n)!=EOF){
init();
for (int i=0;i<n;i++){
for (int j=0;j<n;j++){
scanf("%d",&arr[i][j]);
total[i][j]=arr[i][j];
if (i>0){
total[i][j]+=total[i-1][j];
}
}
}
int ans = getArr();
printf("%d\n",ans);
}
return 0;
}