day51买卖股票的最佳时机含冷冻期&含手续费

力扣309.买卖股票的最佳时机含冷冻期

题目链接:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown/

思路

例如[买,卖,冷冻期,不操作,不操作,买,卖]

1.dp数组的含义:

(1)dp[i][0]:持有状态

(2)dp[i][1]:保持卖出的状态,如例子中不操作就是保持卖出的状态

(3)dp[i][2]:具体卖出的状态,就是卖的那一天

(4)dp[i][3]:冷冻期

2.递推公式:

(1)dp[i][0]持有状态=max(前一天就持有,前一天保持卖出的状态并在今天买入,前一天是冷冻期并在今天买入)

(2)dp[i][1]保持卖出=max(前一天就是卖出状态,前一天是冷冻期)

(3)dp[i][2]具体卖出的状态=前一天持有

(4)dp[i][3]冷冻期=前一天是具体卖出的状态

3.初始化:

dp[0][0] = -prices[0];
dp[0][1] = 0;
dp[0][2] = 0;
dp[0][3] = 0;

需要注意在第一天时,状态1和3是非法状态,那就要在递推公式中推导两者的值。

4.遍历顺序:从前向后

5.打印数组:最后返回的是max(保持卖出状态,具体卖出状态,冷冻期)。不要忘了冷冻期也是可以的。

完整代码

class Solution {
    public int maxProfit(int[] prices) {
        int len = prices.length;
        int[][] dp = new int[len][4];
        dp[0][0] = -prices[0];
        dp[0][1] = 0;
        dp[0][2] = 0;
        dp[0][3] = 0;
        for (int i = 1; i < len; i++) {
            dp[i][0] = dp[i-1][0] > dp[i-1][1]-prices[i] ? dp[i-1][0] : dp[i-1][1]-prices[i];
            dp[i][0] = dp[i][0] > dp[i-1][3]-prices[i] ? dp[i][0] : dp[i-1][3]-prices[i];
            dp[i][1] = Math.max(dp[i-1][1],dp[i-1][3]);
            dp[i][2] = dp[i-1][0]+prices[i];
            dp[i][3] = dp[i-1][2];
        }
        int max = dp[len-1][2] > dp[len-1][1] ? dp[len-1][2] : dp[len-1][1];
        max = max > dp[len-1][3] ? max : dp[len-1][3];
        return max;
    }
}

力扣714.买卖股票的最佳时机含手续费

题目链接:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-transaction-fee/

思路

本题和买卖股票的最佳时机2是一样的,只要在卖出股票的时候加上手续费即可。

完整代码

class Solution {
    public int maxProfit(int[] prices, int fee) {
        int[][] dp = new int[prices.length][2];
        dp[0][0] = -prices[0];
        dp[0][1] = 0;
        for (int i = 1; i < prices.length; i++) {
            dp[i][0] = Math.max(dp[i-1][0],dp[i-1][1]-prices[i]);
            dp[i][1] = Math.max(dp[i-1][1],dp[i-1][0]+prices[i]-fee);
        }
        return dp[prices.length-1][1];
    }
}
好的,以下是使用上述trade_day来交易股票池中的股票的代码实现: ```python import tushare as ts pro = ts.pro_api('你的token') # 这里需要填写你的tushare token # 股票池 stock_pool = ['600519.SH', '000858.SZ', '601318.SH', '000651.SZ', '601888.SH'] # 获取股票交易日历 cal = pro.trade_cal(exchange='', start_date='20160101', end_date='20211231') trade_days = cal[cal.is_open == 1]['cal_date'].values # 遍历所有交易日 for i, trade_day in enumerate(trade_days): if i >= 1500: break # 获取当日所有股票行情数据 df = pro.daily(trade_date=trade_day) # 进行股票交易 for code in stock_pool: # 获取股票当日行情数据 stock_data = df[df.ts_code == code] if len(stock_data) == 0: continue # 判断是否买入股票 if stock_data.iloc[0]['pct_chg'] > 0: print(trade_day, code, '买入') # 判断是否卖出股票 elif stock_data.iloc[0]['pct_chg'] < 0: print(trade_day, code, '卖出') # 否则不进行交易 else: pass ``` 这段代码与之前的代码基本相同,只是在进行股票交易的部分进行了修改。我们首先定义了一个股票池,然后在每个交易日中遍历股票池中的所有股票。对于每个股票,我们从当日所有股票的行情数据中筛选出该股票的行情数据,然后根据涨跌幅判断是否买入或卖出该股票,或者不进行交易。如果买入或卖出,则打印出交易日股票代码和交易动作。注意,这里同样需要用到tushare的API,所以需要先在tushare官网上注册账号并获取token,然后将token替换代码中的相应部分。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值