【力扣】——可获得的最大点数(滑动窗口)
几张卡牌 排成一行,每张卡牌都有一个对应的点数。点数由整数数组 cardPoints 给出。
每次行动,你可以从行的开头或者末尾拿一张卡牌,最终你必须正好拿 k 张卡牌。
你的点数就是你拿到手中的所有卡牌的点数之和。
给你一个整数数组 cardPoints 和整数 k,请你返回可以获得的最大点数。
示例 1:
输入:cardPoints = [1,2,3,4,5,6,1], k = 3
输出:12
解释:第一次行动,不管拿哪张牌,你的点数总是 1 。但是,先拿最右边的卡牌将会最大化你的可获得点数。最优策略是拿右边的三张牌,最终点数为 1 + 6 + 5 = 12 。
示例 2:
输入:cardPoints = [1,79,80,1,1,1,200,1], k = 3
输出:202
提示:
1 <= cardPoints.length <= 10^5
1 <= cardPoints[i] <= 10^4
1 <= k <= cardPoints.length
滑动窗口是一种解决数组/字符串子数组或子串问题的有效方法。
当问题设计到子串问题且长度不变时,这时可以用到滑动窗口的想法大大减小代码难度。(就是控制起始位置和结束位置哪)
#include <iostream>
#include <vector>
#include <algorithm>
#include<bits/stdc++.h>
using namespace std;
int maxScore(vector<int>& cardPoints, int k) {
int totalSum = 0;
int windowSize = cardPoints.size() - k;
int windowSum = 0;
int minWindowSum = INT_MAX;
for (int i = 0; i < cardPoints.size(); i++) {
totalSum += cardPoints[i];
}
for (int i = 0; i < windowSize; i++) {
windowSum += cardPoints[i];
}
minWindowSum = min(minWindowSum, windowSum);
for (int i = windowSize; i < cardPoints.size(); i++) {
windowSum = windowSum - cardPoints[i - windowSize] + cardPoints[i];
minWindowSum = min(minWindowSum, windowSum);
}
return totalSum - minWindowSum;
}
int main() {
int n, k;
cin >> n >> k;
vector<int> cardPoints(n);
for (int i = 0; i < n; i++) {
cin >> cardPoints[i];
}
// 计算最大点数
int maxPoints = maxScore(cardPoints, k);
cout << maxPoints << endl;
return 0;
}