最大和
时间限制:1000 ms | 内存限制:65535 KB
难度:5
描述
给定一个由整数组成二维矩阵(r*c),现在需要找出它的一个子矩阵,使得这个子矩阵内的所有元素之和最大,并把这个子矩阵称为最大子矩阵。
例子:
0 -2 -7 0
9 2 -6 2
-4 1 -4 1
-1 8 0 -2
其最大子矩阵为:
9 2
-4 1
-1 8
其元素总和为15。
输入
第一行输入一个整数n(0< n<=100),表示有n组测试数据;
每组测试数据:
第一行有两个的整数r,c(0< r,c<=100),r、c分别代表矩阵的行和列;
随后有r行,每行有c个整数;
输出
输出矩阵的最大子矩阵的元素之和。
样例输入
1
4 4
0 -2 -7 0
9 2 -6 2
-4 1 -4 1
-1 8 0 -2
样例输出
15
分析:
题目要求最大子矩阵之和。我们都写过一维的最大子串和问题。
一维最大子串和O(n)算法:
int max_sub(int a[],int n)
{
int t=0,maxans=-100000;
for(int i=0;i<n;i++)
{
if(t<0) t=0;
t+=a[i];
if(t>maxans)
maxans=t;
}
return maxans;
}
那么这有什么帮助了?要是我们能把二维问题转化为多个一维最大和就可以轻松的解决了。
我们将从第r行到第k行的每一行中相同列的加起来,可以得到一个一维数组如下:
(ar1+……+ak1, ar2+……+ak2, ……,arn+……+akn)
由此我们可以看出最后所求的就是此一维数组的最大子断和问题,到此我们已经将问题转化为上面的已经解决了的问题了
以第一行为例:我们找所有包含第一行的子矩阵。
在图中,每个框框中的元素之和可以看成某一维数组的一个元素。所以我们将他转换成了4个一维数组。同理从第2,3,。。n行开始,也可以将二维数组转换为一维数组。
AC代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
const int maxn = 110;
int a[maxn][maxn];
int ans;
int n,m;
void max_sub(int k)
{
int t=0;
for(int i=1;i<=m;i++)
{
if(t<0) t=0;
t+=a[k][i];
if(t>ans)
ans=t;
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
scanf("%d",&a[i][j]);
}
ans=a[1][1];
for(int i=1;i<=n;i++)
{
max_sub(i);
for(int j=i+1;j<=n;j++)
{
for(int c=1;c<=m;c++)
a[i][c]+=a[j][c];
max_sub(i);
}
}
printf("%d\n",ans);
}
return 0;
}