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

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

原创
作者头像
福大大架构师每日一题
修改2021-06-16 10:28:25
5480
修改2021-06-16 10:28:25
举报
文章被收录于专栏:福大大架构师每日一题

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编写。代码如下:

代码语言:txt
复制
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
    }
}

执行结果如下:

图片
图片

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

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

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

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

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