蓝桥杯-最大子阵 动态规划

本文介绍了一种求解矩阵最大子矩阵和的动态规划算法。通过将二维矩阵问题转化为一维数组处理,实现了O(n^3)的时间复杂度。文章详细解释了算法原理,并给出了完整的C++代码实现。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目大意:给定一个矩阵,求出它的子矩阵中元素之和的最大值

题目分析:这道题明显是用 动态规划来做。我们可以先分析一维数组的情况,num[] = {1,-3,5,2,-6,9},考虑类似这样的数组,要求出它的最长连续子序列和的最大值。我们从左往右扫描,用dp[i]表示以第i个元素为末尾元素的连续子序列和的最大值。也就是第i个元素一定要包括在内。那么,我们可以得到dp[i] = max(dp[i-1]+num[i],num[i]).  也就是说,如果dp[i-1]大于0,那么dp[i]就等于dp[i-1]再加上num[i];若dp[i-1]<=0, 那么dp[i] = num[i]. 因为前面的值是负值,加上dp[i-1]只会使得dp[i]变小,而我要求的是包含num[i]的最大值。

回到该问题上,因为这里由一维数组变成了矩阵,所以我们尽量把矩阵的最大子阵问题转化为一维的情况。假设矩阵是2*m的大小,如下图所示:

A11A12A13A14A15A16A17A18A19
A21A22A23A24A25A26A27A28A29

我们要求该矩阵的最大子阵,我们可以怎么求? 我们会先处理第一行元素的最大连续子序列和、再处理第二行最大连续子序列和,这时最大子阵是1*m的形式,然后还要考虑2*m的形式,比如可能A14,A24,A15,  A25可以成为最大子阵。 这种情况我们可以采用压缩法,即把每列元素累加起来,变成求 A11+A21, A12+A22, A13+A23, A14+A24, A15+A25,....A19+A29 这个序列的最大连续子序列的和。

如果我们处理3*m的情况,如图所示:

A11A12...      
A21A22...      
A31A32...      

我们可以依次处理第一行、第一二行压缩之后、第一二三行压缩之后、第二行、第二三行压缩后、第三行的一维数组的最大连续子序列和,它们的最大值就是矩阵的最大子阵之和。

当矩阵变成n*m时,两层循环来控制要压缩的第i行和第j行之间的元素,然后用O(n)的时间来求出一维数组的最大连续子序列和,总时间复杂度是O(n^3)

代码展示:

#include <iostream>
#include <algorithm>
#include <string.h>
using namespace std;

int main(){
    int n,m;
    cin>>n>>m;
    int num[510][510];
    int dp[510];
    int temp[510];
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>num[i][j];
        }
    }
    int result = num[0][0];    //先将其设为第一个数
    for(int i=0;i<n;i++){
        memset(temp,0,sizeof(temp));
        for(int j=i;j<n;j++){
            for(int k=0;k<m;k++){
                temp[k] += num[j][k];
            }
            memset(dp,0,sizeof(dp));
            //对temp一维数组进行求最大子序列和
            dp[0] = temp[0];
            for(int u=1;u<m;u++){
                dp[u] = max(dp[u-1]+temp[u],temp[u]);
                if(dp[u]>result)
                    result = dp[u];
            }
        }
    }
    cout<<result<<endl;
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值