二维矩阵中求和最大的子矩阵
- 这一天你来到了蓝桥杯的考场,你发现考场是一个N*M的矩阵。
- 因为你的群友很多,你知道考场内每个人有多强,并且把实力换算成了数值。(因为有的人太弱了,所以可能出现实力值是负数的可能)
- 你想知道考场内实力总和最大的矩阵区域的实力和是多少。
- (注意:区域是按照矩形划分的)
输入描述:
第一行两个整数 N M \left(1\leq N \times M \leq 2e5\right)(1≤N×M≤2e5)
第二到N+1行是一个N*M的矩阵代表考场内的情况 \left( -200<=实力值<=200 \right)(−200<=实力值<=200)
输出描述:
请输出考场内实力总和最大的矩阵区域的实力和是多少
输入实例:
3 2
8 9
10 11
-4 11
输出描述:
45
基本思路:
- 我们先对每一行进行前缀和处理;
- 完成之后,直接进入三重循环:
- 第一重循环 , 确定一个子矩阵的左边界
- 第二重循环 , 确定一个子矩阵的右边界
- 第三重循环 , 从第一行遍历到第n行,用dp数组来记录下当前左右边界区间和的情况,如果是个负数,直接不要累加了,如果是整数直接累加,累加的过程中,我们不断的进行最大值的更新,就可以达到一个矩阵内子矩阵的最大值。
代码展示:
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 210000;
LL dp[N];
int main(){
int n,m;
cin >> n >> m;
vector<vector<LL > > s(n+1 , vector<LL >(m+1 , 0));
for(int i=1;i<=n;i++) {
for(int j=1;j<=m;j++){
cin >> s[i][j];
s[i][j] += s[i][j-1];
}
}
LL ans = 0;
for(int l=1;l<=m;l++){
for(int r = l;r<=m;r++){
for(int i=1;i<=n;i++){
dp[i] = s[i][r] - s[i][l-1];
if(dp[i-1]>0)dp[i]+=dp[i-1];
ans = max(ans , dp[i]);
}
}
}
cout << ans << endl;
}