Maximum Points You Can Obtain from Cards (M)
题目
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.
Example 4:
Input: cardPoints = [1,1000,1], k = 1
Output: 1
Explanation: You cannot take the card in the middle. Your best score is 1.
Example 5:
Input: cardPoints = [1,79,80,1,1,1,200,1], k = 3
Output: 202
Constraints:
1 <= cardPoints.length <= 10^5
1 <= cardPoints[i] <= 10^4
1 <= k <= cardPoints.length
题意
从给定数组中取出k个数,每次只能从头部或者尾部取出,求这k个数的最大和。
思路
因为每次都从头部或尾部取出,剩下的一定是一个连续子数组,问题就可已转化为求这个连续子数组的最小和。
代码实现
Java
class Solution {
public int maxScore(int[] cardPoints, int k) {
int len = cardPoints.length - k;
int totalSum = 0;
for (int num : cardPoints) {
totalSum += num;
}
if (len == 0) return totalSum;
int left = 0, right = 0;
int localSum = 0;
while (len > 0) {
localSum += cardPoints[right++];
len--;
}
int min = localSum;
while (right < cardPoints.length) {
localSum += cardPoints[right++];
localSum -= cardPoints[left++];
min = Math.min(min, localSum);
}
return totalSum - min;
}
}