题目描述
有一个正整数和负整数组成的NxN矩阵,请编写代码找出元素总和最大的子矩阵。请尝试使用一个高效算法。
给定一个int矩阵mat和矩阵的阶数n,请返回元素总和最大的子矩阵的元素之和。保证元素绝对值小于等于100000,且矩阵阶数小于等于200。
测试样例:
[[1,2,-3],[3,4,-5],[-5,-6,-7]],3
返回:10
基本思想:将二维最大和转换为以为最大子和
class SubMatrix {
public:
int maxsum(vector<int>& data,int n){
int result=data[0];
int sum=data[0];
for(int i=1;i<n;++i){
if(sum<0)sum=data[i];
else sum+=data[i];
result=max(result,sum);
}
return result;
}
int sumOfSubMatrix(vector<vector<int> > mat, int n) {
// write code here
if(n==0)return 0;
int result=numeric_limits<int>::min();
for(int i=0;i<n;++i){
vector<int> temp=mat[i];
result=max(result,maxsum(temp,n));
for(int j=i+1;j<n;++j){
for(int k=0;k<n;++k){
temp[k]+=mat[j][k];
}
result=max(result,maxsum(temp,n));
}
}
return result;
}
};