当前位置：首页 > news >正文

LeetCode 3652.按策略买卖股票的最佳时机：滑动窗口

news 2026/6/30 19:33:03

【LetMeFly】3652.按策略买卖股票的最佳时机：滑动窗口

力扣题目链接：https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-using-strategy/

给你两个整数数组prices和strategy，其中：

prices[i]表示第i天某股票的价格。
strategy[i]表示第i天的交易策略，其中：
- -1表示买入一单位股票。
- 0表示持有股票。
- 1表示卖出一单位股票。

同时给你一个偶数整数k，你可以对strategy进行最多一次修改。一次修改包括：

选择strategy中恰好k个连续元素。
将前k / 2个元素设为0（持有）。
将后k / 2个元素设为1（卖出）。

利润定义为所有天数中strategy[i] * prices[i]的总和。

返回你可以获得的最大可能利润。

注意：没有预算或股票持有数量的限制，因此所有买入和卖出操作均可行，无需考虑过去的操作。

示例 1：

输入：prices = [4,2,8], strategy = [-1,0,1], k = 2

输出：10

解释：

修改	策略	利润计算	利润
原始	[-1, 0, 1]	(-1 × 4) + (0 × 2) + (1 × 8) = -4 + 0 + 8	4
修改 [0, 1]	[0, 1, 1]	(0 × 4) + (1 × 2) + (1 × 8) = 0 + 2 + 8	10
修改 [1, 2]	[-1, 0, 1]	(-1 × 4) + (0 × 2) + (1 × 8) = -4 + 0 + 8	4

因此，最大可能利润是 10，通过修改子数组[0, 1]实现。

示例 2：

输入：prices = [5,4,3], strategy = [1,1,0], k = 2

输出：9

解释：

修改	策略	利润计算	利润
原始	[1, 1, 0]	(1 × 5) + (1 × 4) + (0 × 3) = 5 + 4 + 0	9
修改 [0, 1]	[0, 1, 0]	(0 × 5) + (1 × 4) + (0 × 3) = 0 + 4 + 0	4
修改 [1, 2]	[1, 0, 1]	(1 × 5) + (0 × 4) + (1 × 3) = 5 + 0 + 3	8

因此，最大可能利润是 9，无需任何修改即可达成。

提示：

2 <= prices.length == strategy.length <= 10⁵
1 <= prices[i] <= 10⁵
-1 <= strategy[i] <= 1
2 <= k <= prices.length
k是偶数

解题方法：滑动窗口

既然修改范围是定长的，并且最多修改1次，那么就从前往后将每一种修改可能都试试呗。

初始先计算原数组不修改时收益，再从前往后依次尝试修改区间，取收益最大的一个作为答案。

如何从一个区间快速计算出下一个区间呢？变化的有3个：（变化前的）区间起点、区间中点、区间终点，把这三个位置的值更新一下就好了。

时间复杂度O ( l e n ( p r i c e s ) ) O(len(prices))O(len(prices))
空间复杂度O ( 1 ) O(1)O(1)

AC代码

C++

/* * @LastEditTime: 2025-12-18 18:42:50 */typedeflonglongll;classSolution{public:llmaxProfit(vector<int>&prices,vector<int>&strategy,intk){ll ans=0;intn=prices.size();for(inti=0;i<n;i++){ans+=strategy[i]*prices[i];}ll now=ans;for(inti=0;i<k/2;i++){now+=(0-strategy[i])*prices[i];}for(inti=k/2;i<k;i++){now+=(1-strategy[i])*prices[i];}ans=max(ans,now);for(inti=1;i+k<=n;i++){// i-1: 0->original// i+k/2-1: 1->0// i+k-1: original->1now+=(strategy[i-1]-0)*prices[i-1]+(0-1)*prices[i+k/2-1]+(1-strategy[i+k-1])*prices[i+k-1];ans=max(ans,now);}returnans;}};