问题描述
题目链接:最大子阵
资源限制
时间限制:1.0s 内存限制:256.0MB
问题描述
给定一个n*m的矩阵A,求A中的一个非空子矩阵,使这个子矩阵中的元素和最大。
其中,A的子矩阵指在A中行和列均连续的一块。
输入格式
输入的第一行包含两个整数n, m,分别表示矩阵A的行数和列数。
接下来n行,每行m个整数,表示矩阵A。
输出格式
输出一行,包含一个整数,表示A中最大的子矩阵中的元素和。
样例输入
3 3
-1 -4 3
3 4 -1
-5 -2 8
样例输出
10
样例说明
取最后一列,和为10。
数据规模和约定
对于50%的数据,1<=n, m<=50;
对于100%的数据,1<=n, m<=500,A中每个元素的绝对值不超过5000。
解题思路
求最大子阵的元素和,可以将几行数据合为一行,每次比较每一行的各列数据之和与当前列值的大小,如果每一行的各列数据之和大于当前列的值,则将其赋给tmp,比较tmp与maxsum的大小,将maxsum更新为较大的值,并将tmp作为相加后的操作,继续将后面的列加上来,并进行比较;如果当前列数值较大,则将其值赋给tmp,比较tmp与maxsum的大小,将maxsum更新为较大的值,继续之后的操作。
对于题中的例子,sum[]用来存储每列的和,当输入第一行数据为-1,-4,3时,sum存储的为:
sum | 1 | 2 | 3 |
---|---|---|---|
-1 | -4 | 3 |
然后将第一行的每列数据进行相加并与当前列进行比较。此时, 初始化maxsum=-INF,tmp=0;
第一步:
⑴加上第一列数据-1时,更新为maxsum=-1,tmp=-1;
⑵加上第二列数据-4,由于tmp+sum[2]= -1 + -4 =-5 < sum[2] =-4,因此tmp=sum[2]=-4,maxsum=-1;
⑶加上第三列数据3,更新为tmp=-1,maxsum=3;
第二步:将sum加上第二行数据3,4,-1,sum更新为:
sum | 1 | 2 | 3 |
---|---|---|---|
2 | 0 | 2 |
按照第一行数据依次类推,枚举全部的合并列并进行比较求出最大子阵的元素和,即maxsum的值最大。
具体代码
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
#define INF 0x3f3f3f3f
int a[505][505],sum[505];//a存储原始数据,sum存储每一行的列数和
int main()
{
int n,m;
int maxsum=-INF;//最大子阵元素和
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
scanf("%d",&a[i][j]);
}
}
for(int k=1;k<=n;k++)
{
memset(sum, 0, sizeof(sum));//要加,不然会出错
for(int i=k;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
sum[j]+=a[i][j];//统计每一列的和
}
int tmp=0;
for(int j=1;j<=m;j++)
{
if(tmp+sum[j]<sum[j])
tmp=sum[j];
else
tmp=tmp+sum[j];
maxsum=max(tmp,maxsum);
}
}
}
printf("%d",maxsum);
return 0;
}