题目 1445: 蓝桥杯历届试题-最大子阵
一开始以为可以用二维前缀和做,做完TLE了。
空间换时间
常规做法是四重循环,两重枚举矩阵的左上角或者右上角,两重枚举矩阵的长和宽。
时间复杂度分析: 500500500*500 = 62,500,000,000会超时。
所以分析怎么少一重循环。
a[i][j]表示第j列从第1行到第i行的数的和。
再来一个sum 存储前面几列的所有的和。
这样就可以表示任意一个矩阵。
假设当前在第一列,第一列到第i行的数是正的,那后面看第二列的时候就需要加上第一列的这些正的,用一个sum来暂时存储第一列的正的的和。后面算出第二列后加上。
如果第一列到第i行的数是负的,后面就不能加。置sum = 0然后算第二列。
第一重循环枚举首行,第二重循环枚举尾行,第三重循环枚举列,用[尾行,列] - [首行,列] 得到这一列的和。
当某个中间时刻,再枚举到第k列时,前面的k-1列(首行~尾行的列)的和sum可能为正可能为负。为负就置sum=0.为正就让sum + 第k列的和:a[j][k] - a[i-1][k]
.
核心代码
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
sum = 0;
for(int k=1;k<=m;k++)
{
sum += a[j][k] - a[i-1][k];
if(sum > res) res = sum;
if(sum < 0) sum = 0;
}
}
}
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 5e2+10;
int a[N][N];
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
int temp;
cin>>temp;
a[i][j] = a[i-1][j] + temp;
}
int res = -0x3f3f3f3f;
int sum;
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
sum = 0;
for(int k=1;k<=m;k++)
{
sum += a[j][k] - a[i-1][k];
if(sum > res) res = sum;
if(sum < 0) sum = 0;
}
}
}
cout<<res;
}