题目链接:To The Max
给定一个二维矩阵,要求一个子矩阵,使该子矩阵的元素之和是原矩阵所有子矩阵中元素和最大的一个,要求输出该最大和的值。
这个题是最大子段和的扩展,对普通子段和问题,使用dp算法有:
dp[i] = max(dp[i - 1],0) + a[i];//a[i]表示原数组第i个元素,dp[i]表示以i结尾的子段可达的最大值;
然后遍历dp[]数组,找到最大值就是结果;
本题将该问题拓展到矩阵,但可采用同样的思路使:
dp[k] = max(dp[k - 1],0) + sum[i][k] - sum[j][k]; // 1 <= j < i <= n, k <= n;
其中sum[i][k] 表示矩阵第k列1...i行的加和,这样dp[k]就表示矩阵第j...i行中以第k列为结尾的子矩阵可达的最大元素和,最后遍历dp数组取最大值即可(具体实现见代码);
AC代码:
#include <stdio.h>
#define MAX(a,b) ((a) > (b) ? (a) : (b))
#define N 110
int a[N][N]; //和矩阵
int n;
int dp[N];
int main()
{
int i,j,k;
int ans;
while(scanf("%d",&n) != EOF){
for(i = 1;i <= n;i ++){
for(j = 1;j <= n;j ++){
scanf("%d",&a[i][j]);
a[i][j] += a[i - 1][j];
}
}
for(i = 0;i < n ;i ++){
for(j = i + 1;j <= n;j ++){
dp[1] = a[j][1] - a[i][1];
if(i == 0 && j == 1) ans = dp[1];
else ans = MAX(ans,dp[1]);
for(k = 2;k <= n;k ++){
dp[k] = MAX(dp[k - 1] + a[j][k] - a[i][k],a[j][k] - a[i][k]);
ans = MAX(ans,dp[k]);
}
}
}
printf("%d\n",ans);
}
return 0;
}