前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >2021-06-15:返回一个二维数组中,子矩阵最大累加和。

2021-06-15:返回一个二维数组中,子矩阵最大累加和。

作者头像
福大大架构师每日一题
发布2021-08-05 16:09:36
3050
发布2021-08-05 16:09:36
举报

2021-06-15:返回一个二维数组中,子矩阵最大累加和。

福大大 答案2021-06-15:

根据昨天的每日一题计算出0 ~ 0行,0 ~ 1行,0 ~ 2行,……0~N行的子数组最大累加和。

根据昨天的每日一题计算出1 ~ 1行,1 ~ 2行,1 ~ 3行,……1~N行的子数组最大累加和。

根据昨天的每日一题计算出2 ~ 2行,2 ~ 3行,2 ~ 4行,……2~N行的子数组最大累加和。

……

最后取最大值做返回值。

时间复杂度:O(N^2 * M)。

代码用golang编写。代码如下:

代码语言:javascript
复制
package main

import (
    "fmt"
    "math"
)

func main() {
    m := [][]int{{1, 2}, {3, -40}}
    ret := maxSum(m)
    fmt.Println(ret)
}

func maxSum(m [][]int) int {
    if len(m) == 0 || len(m[0]) == 0 {
        return 0
    }
    // O(N^2 * M)
    N := len(m)
    M := len(m[0])
    max := math.MinInt64
    for i := 0; i < N; i++ {
        // i~j
        s := make([]int, M)
        for j := i; j < N; j++ {
            for k := 0; k < M; k++ {
                s[k] += m[j][k]
            }
            max = getMax(max, maxSubArray(s))
        }
    }
    return max
}

func maxSubArray(arr []int) int {
    if len(arr) == 0 {
        return 0
    }
    N := len(arr)
    ans := arr[0]
    pre := arr[0]
    for i := 1; i < N; i++ {
        pre = getMax(arr[i], pre+arr[i])
        ans = getMax(ans, pre)
    }
    return ans
}

func getMax(a int, b int) int {
    if a > b {
        return a
    } else {
        return b
    }
}

执行结果如下:

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2021-06-15,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 福大大架构师每日一题 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档