1850. 捡苹果

1850. 捡苹果

 

 

Alice 和 Bob 在一个漂亮的果园里面工作，果园里面有N棵苹果树排成了一排，这些苹果树被标记成1 - N号。

Alice 计划收集连续的K棵苹果树上面的所有苹果，Bob计划收集连续的L棵苹果树上面的所有苹果。

Alice和Bob选择的区间不可以重合，你需要返回他们能够最大收集的苹果数量。

样例

示例 1:

输入:

A = [6, 1, 4, 6, 3, 2, 7, 4]

K = 3

L = 2

输出: 24

解释：因为Alice 可以选择3-5颗树然后摘到4 + 6 + 3 = 13 个苹果， Bob可以选择7-8棵树然后摘到7 + 4 = 11个苹果，因此他们可以收集到13 + 11 = 24。

示例 2:

输入:

A = [10, 19, 15]

K = 2

L = 2

输出: -1

解释：因为对于 Alice 和 Bob 不能选择两个互不重合的区间。

注意事项

• N 是整数且在以下范围内：[2，600]
• K 和 L 都是整数且在以下范围内：[1，N-1]
• 数组A的每个元素都是以下范围内的整数：[1，500]

 

 

public class Solution {

     public int PickApples(int[] A, int K, int L) {

            int length=K+L;

            if (length>A.length){

                return -1;

            }

            int max=getMax(A,K,L);

            int max2=getMax(A,L,K);

            return Math.max(max,max2);

        }

 

 

        private int getMax(int[] A, int K, int L) {

            int max=0;

            int[] dpR=new int[A.length];

            int[] dpL=new int[A.length];

            int sumR=0;

            int sumL=0;

            for (int i=0; i<K; i++){

                sumR+=A[i];

            }

            dpR[K-1]=sumR;

            int lengthR=A.length-L;

            for (int i = K; i < lengthR; i++) {

                sumR-=A[i-K];

                sumR+=A[i];

                dpR[i]=Math.max(dpR[i-1],sumR);

            }

            for (int i=A.length-1; i>=lengthR; i--){

                sumL+=A[i];

            }

            dpL[A.length-L]=sumL;

            for (int i = lengthR-1; i>=K; i--) {

                sumL-=A[i+L];

                sumL+=A[i];

                dpL[i]=Math.max(dpL[i+1],sumL);

            }

            for (int i = K-1; i <lengthR; i++) {

                max=Math.max(max,dpR[i]+dpL[i+1]);

            }

            return max;

        }

 

 

}