题意: 给出矩阵,求最大子段和。
思路:
给定一个数据
5 4
11 18 24 -13 24
-25 -12 -13 16 -11
-9 -6 11 -6 14
19 24 6 -11 -2
其中m代表列,n代表行(m先输入)
以行或列为基准(这里以列为基准, 并把每行的i~j列和相加存入一个数组tmp[])
解释一下前两个for循环
for(int i = 1; i<=m; i++)
Arrays.fill(tmp, 0);
for(int j = i; j<=m; j++)
for(int k = 1; k<=n; k++)
tmp[k] = (long)(tmp[k] + a[k][j]);
i = 1时 j可以选择 1,2,3,4,5列
i =1, j=1 tmp[每行1, 2, 3, 4] = 每行i~j列的和(1~1列)
i=1, j=2 tmp[每行] = 每行i~j列的和(1~2列)
i=2, j=5 tmp[每行] = 每行i~j列的和(2~5列)
其实发现 在每次i=数, j=数中, 都把i~j列的和转换成了一个一维数组tmp[行1,行2,行3,行4]. 就可以用求 最大子序列和 的方式来求 最大的子矩阵了。
c++代码 链接
https://blog.youkuaiyun.com/zchahaha/article/details/51465376
Java代码
import java.util.Arrays;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int m = sc.nextInt(), n = sc.nextInt();
int a[][] = new int[n+1][m+1];
for(int i = 1; i<=n; i++)
for(int j = 1; j<=m; j++)
a[i][j] = sc.nextInt();
long tmp[] = new long[n+1];
long ans = 0;
for(int i = 1; i<=m; i++){
Arrays.fill(tmp, 0);
for(int j = i; j<=m; j++){
for(int k = 1; k<=n; k++)
tmp[k] = (long)(tmp[k] + a[k][j]);
// System.out.print("i:"+ i+" j:"+j+ " "); // 测试
// output(tmp);
long v = 0;
for(int k = 1; k<=n; k++)
{
v = (long)(v + tmp[k]);
if(v > 0 && v > ans)
ans = v;
if(v < 0)
v = 0;
}
}
}
System.out.println(ans);
}
static void output(long a[]){
for(int i = 1; i<a.length; i++)
System.out.print(a[i]+" ");
System.out.println();
}
static int max(int x, int y){
return x>y?x:y;
}
}