又到了吃午饭的时间,你和你的同伴刚刚研发出了最新的GSS-483型自动打饭机器人,现在你们正在对机器人进行功能测试。
为了简化问题,我们假设午饭一共有N个菜,对于第i个菜,你和你的同伴对其定义了一个好吃程度(或难吃程度,如果是负数的话……)A[i],
由于一些技(经)术(费)限制,机器人一次只能接受一个指令:两个数L, R——表示机器人将会去打第L~R一共R-L+1个菜。
本着不浪费的原则,你们决定机器人打上来的菜,含着泪也要都吃完,于是你们希望机器人打的菜的好吃程度之和最大
然而,你善变的同伴希望对机器人进行多次测试(实际上可能是为了多吃到好吃的菜),他想知道机器人打M次菜能达到的最大的好吃程度之和
当然,打过一次的菜是不能再打的,而且你也可以对机器人输入-1, -1,表示一个菜也不打
import java.util.Scanner;
public class Main{
public static int max(int a, int b) {
return a>b?a:b;
}
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int n = scan.nextInt();
int m = scan.nextInt();
scan.nextLine();
int [] A = new int[100009];
int [][] dp= new int[m+9][n+9];
for(int i=1; i<=n;) {
A[i++] = scan.nextInt();
}
for(int i=0; i<n+9; i++)
dp[0][i] = 0;
int pre;
int max = Integer.MIN_VALUE;
for(int i=1; i<=m; i++) {
pre= Integer.MIN_VALUE;
for(int j=i; j<=n; j++) {
pre = max(pre, dp[i-1][j-1]);
// 如果需要求四段和, 那么在当前位置的元素有两种选择
// 1) 接到左边的连续段中
// 2) 自起一段 如果当前位置是i,且当前段数是第四段 那就代表,第四段从index为i的地方开始, 同时它需要在之前的字段中寻找最大值做连接。
dp[i][j] = max(pre, dp[i][j-1]) + A[j];
if(max<dp[i][j])
max=dp[i][j];
}
}
System.out.println(max);
}
}
这样解 通过80%案例,给出提示是检查是否有越界情况,在eclipse下测试一下,当m,n取值很大时,会给出错误
java.lang.OutOfMemoryError: Java heap space
改用滚动数组
import java.util.Scanner;
public class Main{
public static int max(int a, int b) {
return a>b?a:b;
}
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
int n = scan.nextInt();
int m = scan.nextInt();
scan.nextLine();
int [] A = new int[100009];
int [][] dp= new int[2][n+9];
for(int i=1; i<=n;) {
A[i++] = scan.nextInt();
}
for(int i=0; i<n+9; i++)
dp[0][i] = 0;
int pre;
int max = Integer.MIN_VALUE;
for(int i=1; i<=m; i++) {
pre= Integer.MIN_VALUE;
for(int j=i; j<=n; j++) {
pre = max(pre, dp[(i-1)&1][j-1]);
dp[i&1][j] = max(pre, dp[i&1][j-1]) + A[j];
if(max<dp[i&1][j])
max=dp[i&1][j];
}
}
System.out.println(max);
}
}
这样没有越界错误,但是在时间复杂度上有问题。