题目描述:
几张卡牌 排成一行,每张卡牌都有一个对应的点数。点数由整数数组 cardPoints 给出。
每次行动,你可以从行的开头或者末尾拿一张卡牌,最终你必须正好拿 k 张卡牌。
你的点数就是你拿到手中的所有卡牌的点数之和。
给你一个整数数组 cardPoints 和整数 k,请你返回可以获得的最大点数。
思路:
滑动窗口:
每次只能取最左边或最右边上的数,最后剩下的只能是中间连续的一段,只要中间剩下的一段的值的和是最小的,那么两边取到的就是最大值
代码如下:
C++:
class Solution {
public:
int maxScore(vector<int>& cardPoints, int k) {
int sum=0;
int res=INT_MAX;
for(int cardPoint:cardPoints){
sum+=cardPoint;
}
if(k==cardPoints.size()) return sum;
int remainLength=cardPoints.size()-k;
int temp=0;
for(int i=0;i<cardPoints.size();i++){
temp+=cardPoints[i];
if(i>=remainLength){
temp-=cardPoints[i-remainLength];
}
if(i>=remainLength-1){
res=min(res,temp);
}
}
return sum-res;
}
};
Python:
class Solution:
def maxScore(self, cardPoints: List[int], k: int) -> int:
if len(cardPoints)==k:
return sum(cardPoints)
resMove = sum(cardPoints[:len(cardPoints)-k])
current = resMove
left_idx,right_idx=0,len(cardPoints)-k-1
for _ in range(k):
left_idx+=1
right_idx+=1
current = current-cardPoints[left_idx-1]+cardPoints[right_idx]
resMove=min(resMove,current)
return sum(cardPoints)-resMove