一个M*N的矩阵,找到此矩阵的一个子矩阵,并且这个子矩阵的元素的和是最大的,输出这个最大的值。
例如:3*3的矩阵:
-1 3 -1
2 -1 3
-3 1 2
和最大的子矩阵是:
3 -1
-1 3
1 2
Input
第1行:M和N,中间用空格隔开(2 <= M,N <= 500)。 第2 - N + 1行:矩阵中的元素,每行M个数,中间用空格隔开。(-10^9 <= M[i] <= 10^9)
Output
输出和的最大值。如果所有数都是负数,就输出0。
Input示例
3 3 -1 3 -1 2 -1 3 -3 1 2
Output示例
7
问题分析:
首先将二维压缩变为一维,然后用类似于求最大子段和的方法DP求解即可。
代码如下:
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <cstring>
using namespace std;
const int MAXN = 505;
const int MAX = 1e9 + 5;
int down[MAXN], dp[MAXN];
int square[MAXN][MAXN];
int main()
{
int M, N;
scanf("%d %d", &M, &N);//注意N是行数,M是列数
for(int i = 0; i < N; i++)
for(int j = 0; j < M; j++)
scanf("%d", &square[i][j]);
int ans = -MAX;//ans初始化最小值
for(int i = 0; i < N; i++){//i, j两重循环将行数压缩,达到降维的目的
memset(down, 0, sizeof(down));//down为降维后的数组,注意每次清空
for(int j = i; j < N; j++){
for(int k = 0; k < M; k++)
down[k] += square[j][k];
dp[0] = down[0];//给dp一个初始值,用于dp[k - 1]时的求解
if(ans < dp[0]) ans = dp[0];//因为找的是最大值,所以会多次循环,保留最大值
for(int k = 1; k < M; k++){
dp[k] = max(dp[k - 1] + down[k], down[k]);//DP求解
if(ans < dp[k])
ans = dp[k]; //因为找的是最大值,所以会多次循环,保留最大值
}
}
}
printf("%d", ans);
return 0;
}