leetcode 6道股票题 动态规划dp

思路:
①每天对股票都有三个选择:买入、卖出、保持
②dp数组描述的状态有三种:第n天、第k次交易、是否持有股票

状态方程:
i表示持有股票天数,k表示交易的次数, 0表示没有持有股票,1表示持有股票
①今天(第i天) 没有持有股票 2种原因:
     A.昨天(第i-1天)也没有持有股票,保持不变
     B.昨天(第i-1天)持有股票,但是今天卖了,得到prices[i]元
  dp[i][k][0] = max(dp[i-1][k][0], dp[i-1][k][1] + prices[i]);

②今天(第i天) 持有股票 2种原因:
     A.昨天(第i-1天)持有股票,保持不变
     B.昨天(第i-1天)没有持有股票,今天花prices[i]元买入股票
  dp[i][k][1] = max(dp[i-1][k][1], dp[i-1][k-1][0] - prices[i]);

题目:
121. 买卖股票的最佳时机
     给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
     如果你最多只允许完成一笔交易(即买入和卖出一支股票),设计一个算法来计算你所能获取的最大利润。
     注意你不能在买入股票前卖出股票。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(!prices.size())
            return 0;
        
        vector<vector<int>> dp(prices.size()+1, vector<int>(2));

        dp[0][1] = -prices[0], dp[0][0] = 0;
        for(int i=1; i<prices.size(); i++){
            dp[i][0] = max(dp[i-1][0], dp[i-1][1]+prices[i]);
            dp[i][1] = max(dp[i-1][1], -prices[i]);
        }

        return max(dp[prices.size()-1][0], 0);        
    }
};


122. 买卖股票的最佳时机 II
     给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
     设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
     注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(!prices.size())
            return 0;
        
        vector<vector<int>> dp(prices.size()+1, vector<int>(2));

        dp[0][1] = -prices[0], dp[0][0] = 0;
        for(int i=1; i<prices.size(); i++){
            dp[i][0] = max(dp[i-1][0], dp[i-1][1]+prices[i]);
            dp[i][1] = max(dp[i-1][1], dp[i-1][0]-prices[i]);
        }

        return max(dp[prices.size()-1][0], 0);      
    }
};


123. 买卖股票的最佳时机 III
     给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
     设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
     注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size()==0)
            return 0;
            
        vector<vector<vector<int>>> dp(prices.size(), vector<vector<int>>(3, vector<int>(2)));

        dp[0][1][1] = -prices[0];
        dp[0][2][1] = -prices[0];
        
        for(int i=1; i<prices.size(); i++){
            for(int j=1; j<=2; j++){
                dp[i][j][0] = max(dp[i-1][j][0], dp[i-1][j][1] + prices[i]);
                dp[i][j][1] = max(dp[i-1][j][1], dp[i-1][j-1][0] - prices[i]);
            }
        }
        return max(0, dp[prices.size()-1][2][0]);
    }
};


188. 买卖股票的最佳时机 IV
     给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
     设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
     注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(!prices.size())
            return 0;
		//防止k值太大爆栈(当k大于天数的时候,可以看作是无限次买入卖出)
        if(k<prices.size()){
            vector<vector<vector<int>>> dp(prices.size(), vector<vector<int>>(k+1, vector<int>(2)));

            for(int i=1; i<=k; i++)
                dp[0][i][1] = -prices[0];

            for(int i=1; i<prices.size(); i++){
                for(int j=1; j<=k; j++){
                    dp[i][j][0] = max(dp[i-1][j][0], dp[i-1][j][1] + prices[i]);
                    dp[i][j][1] = max(dp[i-1][j][1], dp[i-1][j-1][0] - prices[i]);
                }
            }
            return max(0, dp[prices.size()-1][k][0]);
        }else{
            vector<vector<int>> dp(prices.size(), vector<int>(2));

            dp[0][1] = -prices[0];
            for(int i=1; i<prices.size(); i++){
                dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i]);
                dp[i][1] = max(dp[i-1][1], dp[i-1][0] - prices[i]);
            }
            return max(0, dp[prices.size()-1][0]);
        }
        return 0;
    }
};


309. 最佳买卖股票时机含冷冻期
     给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。​
     设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
          你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
          卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size()<=1)
            return 0;

        vector<vector<int>> dp(prices.size(),vector<int>(2));

        dp[0][1] = -prices[0];
        dp[1][1] = max(dp[0][1], -prices[1]);
        dp[1][0] = max(dp[0][0], dp[0][1] + prices[1]);

        for(int i=2; i<prices.size(); i++){
            dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i]);
            dp[i][1] = max(dp[i-1][1], dp[i-2][0] - prices[i]);
        }

        return max(0, dp[prices.size()-1][0]);
    }
};


714. 买卖股票的最佳时机含手续费
     给定一个整数数组 prices,其中第 i 个元素代表了第 i 天的股票价格 ;非负整数 fee 代表了交易股票的手续费用。
     你可以无限次地完成交易,但是你每次交易都需要付手续费。如果你已经购买了一个股票,在卖出它之前你就不能再继续购买股票了。
     返回获得利润的最大值。

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        if(!prices.size())
            return 0;
        vector<vector<int>> dp(prices.size(), vector<int>(2));

        dp[0][1] = -prices[0] - fee;
        for(int i=1; i<prices.size(); i++){
            dp[i][1] = max(dp[i-1][1], dp[i-1][0]-prices[i]-fee);
            dp[i][0] = max(dp[i-1][0], dp[i-1][1]+prices[i]);
        }

        return max(0, dp[prices.size()-1][0]);
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值