历届试题 最大子阵

时间限制：1.0s   内存限制：256.0MB

提交此题    

问题描述

　　给定一个n*m的矩阵A，求A中的一个非空子矩阵，使这个子矩阵中的元素和最大。

　　其中，A的子矩阵指在A中行和列均连续的一块。

输入格式

　　输入的第一行包含两个整数n, m，分别表示矩阵A的行数和列数。

　　接下来n行，每行m个整数，表示矩阵A。

输出格式

　　输出一行，包含一个整数，表示A中最大的子矩阵中的元素和。

样例输入

3 3

-1 -4 3

3 4 -1

-5 -2 8

样例输出

10

样例说明

　　取最后一列，和为10。

数据规模和约定

　　对于50%的数据，1<=n, m<=50；

　　对于100%的数据，1<=n, m<=500，A中每个元素的绝对值不超过5000。

几乎同样的代码C++能过100%的数据，而java只能过50%的数据，郁闷啊

import java.util.Scanner;

public class 最大子阵 {

static int ans=Integer.MIN_VALUE;
public static void main(String[] args) {

Scanner sc=new Scanner(System.in);

//while(sc.hasNext()){
int n=sc.nextInt();
int m=sc.nextInt();
int[][] data=new int
[m];
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
data[i][j]=sc.nextInt();
}
}
int[] dp=new int[m+1];
//统计几行的
for (int h = 1; h <= n; h++) {
//从哪一行开始算起
for (int i = 0; i < n; i++) {
int[] arr=new int[m+1];
for (int j = i; j < i+h&&j<n; j++) {

for (int p = 1; p < arr.length; p++) {
arr[p]+=data[j][p-1];
}
dp[1]=arr[1];
if(dp[1]>ans){
ans=dp[1];
}
for (int w = 2; w < arr.length; w++) {
if(dp[w-1]<0)
dp[w]=arr[w];
else
dp[w]=dp[w-1]+arr[w];
if(dp[w]>ans)
ans=dp[w];
}
}
//getSum(arr);
}
}
//}
System.out.println(ans);
sc.close();
}

private static void getSum(int[] arr) {
int sum=0,max=Integer.MIN_VALUE;
for (int i = 0; i < arr.length; i++) {
int num=arr[i];

if(sum>0)
sum += num;
else
sum = num;

if(sum>max)
max = sum;
}
if(max>ans)
ans=max;
}
}