Table of contents
Problem:
There are several cards arranged in a row, and each card has an associated number of points. The points are given in the integer array cardPoints
.
In one step, you can take one card from the beginning or from the end of the row. You have to take exactly k
cards.
Your score is the sum of the points of the cards you have taken.
Given the integer array cardPoints
and the integer k
, return the maximum score you can obtain.
Example 1:
Input: cardPoints = [1,2,3,4,5,6,1], k = 3
Output: 12
Explanation: After the first step, your score will always be 1. However, choosing the rightmost card first will maximize your total score. The optimal strategy is to take the three cards on the right, giving a final score of 1 + 6 + 5 = 12.
Example 2:
Input: cardPoints = [2,2,2], k = 2
Output: 4
Explanation: Regardless of which two cards you take, your score will always be 4.
Example 3:
Input: cardPoints = [9,7,7,9,7,7,9], k = 7
Output: 55
Explanation: You have to take all the cards. Your score is the sum of points of all cards.
Constraints:
1 <= cardPoints.length <= 105
1 <= cardPoints[i] <= 104
1 <= k <= cardPoints.length
Solution:
Using two pointer
Algorithm
Result can be k elements from left and 0 elements from right, Or (k-1) elements from left and 1 element from the right and so on .... till 0 elements from left and k elements from right.
- Create a pointer
left
at the start ofarr
and another pointerright
at end ofarr
. Initialize a variablesum
with value 0 to store the sum ofk
elements. - Increment
left
to iterate over the firstk
elements and add each element tosum
. - Initialize integer
result
as sum (ie k elements from left and 0 elements from right). - Now keep on removing one element from
left
and add one element fromright
. At each step compare the resultant sum with the result. Keep maximum of sum and result as result. - Once the
arr
iteration is complete, return theresult
.
class Solution {
public int maxScore(int[] arr, int k) {
int sum = 0;
int left = 0;
int right = arr.length - 1;
for(; left<k; left++) {
sum += arr[left];
}
left--;
int result = sum;
while(left >= 0) {
sum = sum + arr[right] - arr[left];
result = Math.max(result, sum);
right--;
left--;
}
return result;
}
}
Time complexity: O(k)
as we need to iterate over k elements
Space complexity: O(1)
as we are not creating any additional data structure