LeetCode每日一题解析：按策略买卖股票的最佳时机

LeetCode每日一题解析：按策略买卖股票的最佳时机

在LeetCode上，我们经常遇到各种算法和逻辑题。今天，我们将解析一道关于股票买卖策略的题目——3652. 按策略买卖股票的最佳时机。这道题目主要考察的是滑动窗口的应用，通过计算和比较不同策略下的利润，来找到最佳买卖时机。

题目解析

题目要求我们根据给定的股票价格数组 prices 和交易策略数组 strategy，计算在遵守这些策略的情况下，买卖股票能获得的最大利润。其中，strategy 数组中的元素表示在对应天数的交易策略：1表示买入，-1表示卖出，0表示不进行交易。

思路分析

首先，我们需要计算在没有任何限制的情况下，按照原始价格数组的最大利润。然后，我们使用滑动窗口的方法，维护一个长度为 k 的窗口，其中 k 是策略数组的长度。通过移动窗口，我们能够计算在不同策略组合下的利润变化，并从中找出最大利润。

代码实现

以下是该题目的Java代码实现，首先计算原始利润，然后通过滑动窗口计算不同策略组合下的利润变化，最终返回最大利润。

public long maxProfit(int prices, int strategy, int k) {
    int n = prices.length;
    long ans = 0;
    long diff = 0;
    int half = k / 2;
    for (int i = 0; i < half; i++) {
        ans += (long) prices[i] * strategy[i];
        diff -= (long) prices[i] * strategy[i];
    }
    for (int i = half; i < k; i++) {
        ans += (long) prices[i] * strategy[i];
        diff += prices[i] * (1L - strategy[i]);
    }
    long maxDiff = Math.max(0, diff);
    for (int i = k; i < n; i++) {
        ans += (long) prices[i] * strategy[i];
        diff += (long) prices[i - k] * strategy[i - k];
        diff -= prices[i - half];
        diff += prices[i] * (1L - strategy[i]);
        maxDiff = Math.max(maxDiff, diff);
    }
    return ans + maxDiff;
}

优化思路

在初次实现中，我们使用了多个if-else判断来处理不同策略的情况，这导致了代码的执行效率不高。为了优化性能，我们可以去掉这些判断，直接通过计算差值来更新利润。这样，我们能够显著减少代码的执行时间，提高效率。

以上就是关于LeetCode每日一题——3652. 按策略买卖股票的最佳时机的解析和代码实现。希望对大家有所帮助！