程序员代码面试指南刷题--第八章.子矩阵的最大累加和问题

题目描述
给定一个矩阵matrix，其中的值有正、有负、有0，返回子矩阵的最大累加和
例如，矩阵matrix为：
-90 48 78
64 -40 64
-81 - 7 66
其中，最大的累加和的子矩阵为：
48 78
-40 64
-7 66
所以返回累加和209。
例如，matrix为：
-1 -1 -1
-1 2 2
-1 -1 -1
其中，最大累加和的子矩阵为：
2 2
所以返回4
[要求]
时间复杂度为O(n^2m)，空间复杂度为O(nm)

输入描述:

第一行有两个整数N，M。分别表示矩阵的行数/列数
接下来N行，每行M个整数表示矩阵内的数

输出描述:

输出一个整数表示答案

示例1

输入

3 3
-90 48 78
64 -40 64
-81 -7 66

输出

209

解法一：利用一维数组求最大子数组的方法

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] info = br.readLine().trim().split(" ");
        int row = Integer.parseInt(info[0]);
        int col = Integer.parseInt(info[1]);
        int[][] arr = new int[row][col];
        for(int i=0;i<row;i++){
            String[] tmp = br.readLine().trim().split(" ");
            for(int j=0;j<tmp.length;j++){
                arr[i][j] = Integer.parseInt(tmp[j]);
            }
        }
        int res = getRes(arr);
        System.out.println(res);
    }
    public static int getRes(int[][] arr){
        if(arr==null||arr.length==0||arr[0].length==0) return 0;
        int max = Integer.MIN_VALUE;
        for(int i=0;i<arr.length;i++){
            int[] tmp = new int[arr[0].length];
            for(int j=i;j<arr.length;j++){
                int cur = 0;
                for(int k=0;k<arr[0].length;k++){
                    tmp[k] += arr[j][k];
                    cur += tmp[k];
                    max = Math.max(cur,max);
                    cur = cur<=0?0:cur;
                }
            }
        }
        return max;
    }
}