题目:有一个正整数和负整数组成的NxN矩阵,请编写代码找出元素总和最大的子矩阵。请尝试使用一个高效算法。
给定一个int矩阵mat和矩阵的阶数n,请返回元素总和最大的子矩阵的元素之和。保证元素绝对值小于等于100000,且矩阵阶数小于等于200。
测试样例:
[[1,2,-3],[3,4,-5],[-5,-6,-7]],3
返回:10
解析:首先求得当前行的最大连续和,然后把后面的每一行加在当前行后继续求最大连续和,直到最后一行遍历完为止。
import java.util.*;
public class SubMatrix {
//求解数组的最大连续和
public static int getMax(int [] arr){
int max=arr[0];
int sum =arr[0];
for(int i=1;i<arr.length;i++){
sum=Math.max(arr[i],arr[i]+sum);
max=Math.max(sum,max);
}
return max;
}
public int sumOfSubMatrix(int[][] mat, int n) {
int max=Integer.MIN_VALUE;
for(int i=0;i<n;i++){
int [] temp =mat[i];//当前行
max=Math.max(max,getMax(temp));//当前行的最大值
for(int j=i+1;j<n;j++){//循环加上后面的每一行再判断最大值
for(int m=0;m<n;m++){
temp[m]+=mat[j][m];
}
max=Math.max(max,getMax(temp));
}
}
return max;
}
}