买卖股票系列

1.买卖股票的最佳时机

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。

如果你最多只允许完成一笔交易(即买入和卖出一支股票一次),设计一个算法来计算你所能获取的最大利润。

leetcode 121

int maxProfit(vector<int>& prices) {
    int minprice = INT_MAX, maxprofit = 0;
    for (int price: prices) {
        maxprofit = max(maxprofit, price - minprice);
        minprice = min(price, minprice);
    }
    return maxprofit;
}

2.买卖股票的最佳时机II

leetcode 122

给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。

int maxProfit(vector<int>& prices) {
    int ans = 0;
    for (int i = 1; i < prices.size(); ++i) {
        ans += max(prices[i] - prices[i - 1], 0);
    }
    return ans;
}

3.买卖股票的最佳时机III

leetcode 123

给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。

分析

 对于任意一天考虑四个变量:
    fstBuy: 在该天第一次买入股票可获得的最大收益 
    fstSell: 在该天第一次卖出股票可获得的最大收益
    secBuy: 在该天第二次买入股票可获得的最大收益
    secSell: 在该天第二次卖出股票可获得的最大收益
 分别对四个变量进行相应的更新, 最后secSell就是最大
 收益值(secSell >= fstSell)
int maxProfit(vector<int>& prices) {
    int fstBuy = INT_MIN, fstSell = 0;
    int secBuy = INT_MIN, secSell = 0;
    for(int p : prices) {
        fstBuy = max(fstBuy, -p);
        fstSell = max(fstSell, fstBuy + p);
        secBuy = max(secBuy, fstSell - p);
        secSell = max(secSell, secBuy + p); 
    }
    return secSell;
}

4.买卖股票的最佳时机 IV

leetcode 188

给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。

思路dp[i][j]表示最多交易i次时在第j天能获得的最大利润。则:

  • 如果不作处理,第j天的最大利润就和第j-1天相等。
  • 如果卖出股票,设是在第m天买入,那么第j天的最大利润就是两天的价格差+最多交易i-1次时第m天的获利。当然,这里的m需要从0遍历到j-1。

递推公式为:

dp[i][j] = max (dp[i][j-1], prices[j] - prices[m] + d[i-1][m] (m = 0, 1, ... j-1))
  • 时间复杂度:O(n*n*k)。遍历dp的每个元素需要n * k,其中还需要再遍历找到买入天数。
  • 空间复杂度:O(n * k)。考虑dp数组占用的空间。
int maxProfit(int k, vector<int>& prices) {
    int n = prices.size();
    if (n == 0 || k == 0) return 0;
    if (k >= n / 2){
        int res = 0;
        for (int i = 1; i < n; i ++){
            res += max(0, prices[i] - prices[i - 1]);
        }
        return res;
    }
    vector<vector<int>> dp(k + 1, vector<int>(n));
    for (int i = 1; i <= k; ++i) {
        int maxDiff = -prices[0];
        for (int j = 1; j < dp[0].size(); ++j) {
            maxDiff = max(maxDiff, dp[i - 1][j - 1] - prices[j]);
            dp[i][j] = max(dp[i][j - 1], prices[j] + maxDiff);
        }
    }
    return dp[k][n - 1];
}

5.买卖股票的最佳时机含手续费

leetcode 714

给定一个整数数组 prices,其中第 i 个元素代表了第 i 天的股票价格 ;非负整数 fee 代表了交易股票的手续费用。

你可以无限次地完成交易,但是你每笔交易都需要付手续费。如果你已经购买了一个股票,在卖出它之前你就不能再继续购买股票了。

返回获得利润的最大值。

分析

我们维护两个变量 cashhold, 前者表示当我们不持有股票时的最大利润,后者表示当我们持有股票时的最大利润。

第i天时,根据第i-1天状态更新cashhold的值,对于 cash, 可以保持不变,或者将手上的股票卖出,状态转移方程为:
cash = max(cash, hold + prices[i] - fee) 对于 hold, 可以保持不变,或者买入这一天的股票,状态转移方程为: hold = max(hold, cash - prices[i]).

  • 时间复杂度 O(n)
  • 空间复杂度 O(1)
int maxProfit(vector<int>& prices, int fee) {
    int cash = 0, hold = -prices[0];
    for (int i = 1; i < prices.size(); i++) {
        cash = max(cash, hold + prices[i] - fee);
        hold = max(hold, cash - prices[i]);
    }
    return cash;
}

6.最佳买卖股票时机含冷冻期

leetcode 1312

给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):

  • 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
  • 卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。

分析f[i]表示 第i天结束之后 累计最大收益,由于我们最多只能同时买入(持有)一支股票,并且卖出股票后有冷冻期的限制,因此我们会有三种不同的状态:

  • 当前持有一支股票,累计最大收益为 f[i][0]
  • 当前不持有股票,并处于冷冻期中,最大收益为 f[i][1]
  • 当前不持有股票,并且不处于冷冻期中,最大收益为 f[i][2]

这里的「处于冷冻期」指的是在第 ii 天结束之后的状态。也就是说:如果第 i天结束之后处于冷冻期,那么第 i+1 天无法买入股票。

则状态转移方程为:

f[i][0] = max(f[i - 1][0], f[i - 1][2] - prices[i])
f[i][1] = f[i - 1][0] + prices[i]
f[i][2] = max(f[i - 1][1], f[i - 1][2]) 
如果一共有n天,最终答案为:
ans = max(f[n - 1][0], f[n - 1][1], f[n - 1][2])
int maxProfit(vector<int>& prices) {
    int n = prices.size();
    if(!n) return 0;

    vector<vector<int>> f(n, vector<int>(3, 0));
    f[0][0] = -prices[0];

    for (int i = 1; i < n; i++) {
        f[i][0] = max(f[i - 1][0], f[i - 1][2] - prices[i]);
        f[i][1] = f[i - 1][0] + prices[i];
        f[i][2] = max(f[i - 1][1], f[i - 1][2]);
    }
    return max(f[n - 1][1], f[n - 1][2]);
}

空间优化代码:

int maxProfit(vector<int>& prices) {
    int n = prices.size();
    if(!n) return 0;

    int buy = -prices[0], cooldown = 0, sale = 0, tmp, cool_bak;

    for (int i = 1; i < n; i++) {
        tmp = buy;
        buy = max(buy, sale - prices[i]);
        cool_bak = cooldown;
        cooldown = tmp + prices[i];
        sale = max(cool_bak, sale);
        // f[i][0] = max(f[i - 1][0], f[i - 1][2] - prices[i]);
        // f[i][1] = f[i - 1][0] + prices[i];
        // f[i][2] = max(f[i - 1][1], f[i - 1][2]);
    }
    return max(cooldown, sale);
}

打赏一下

取消

感谢您的支持,我会继续努力的!

扫码支持
扫码支持
扫码打赏,你说多少就多少

打开支付宝扫一扫,即可进行扫码打赏哦