企业官网建设流程全解析

【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：

输出：10

解释：

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

示例 2：

输出：9

解释：

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

提示：

• 2 <= prices.length == strategy.length <= 105
• 1 <= prices[i] <= 105
• -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;}};

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

千篇源码题解已开源