java中的最大前缀_AcWing 126. 最大的和 Java 版（贪心 + 前缀和 + 暴力枚举）

贪心 + 前缀和 + 暴力枚举

要找到子矩阵的最大和

先计算每行的前缀和

然后遍历构造子矩阵的开始列、结束列

然后遍历每行，加入局部子矩阵，逐渐构成一个大的子矩阵. (贪心)

时间复杂度：$O(n^3)$

AC Code

import java.util.*;

import static java.lang.System.out;

public class Main{

public static void main(String[] args) {

Scanner in = new Scanner(System.in);

int len = in.nextInt();

int[][] arr = new int[len][len];

for(int i = 0; i < len; i++)

for(int j = 0; j < len; j++) arr[i][j] = in.nextInt();

// 前缀和

int[][] pre = new int[len][len + 1];

for(int i = 0; i < len; i++)

for(int j = 1; j <= len; j++) pre[i][j] += pre[i][j - 1] + arr[i][j - 1];

int ans = Integer.MIN_VALUE;

// start 开始列, end 奇数列, h 每一行局部子矩阵

for(int left = 1; left <= len; left++) {

for(int right = left; right <= len; right++) {

int sum = 0;

for(int height = 0; height < len; height++) {

sum += pre[height][right] - pre[height][left - 1];

if(sum > ans) ans = sum;

// 负贡献， 不要

if(sum < 0) sum = 0;

}

}

}

out.println(ans);

}

}