leetcode 买卖股票的最佳时机 IV (DP)

23 篇文章 0 订阅

题目大意:

给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。

注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

解题思路:

首先,我们列出简单的DP的方法:假设dp[n][k]代表第n天还有k次交易额可以用。那么状态转移为:

dp[n][k] = dp[n-1][k]   //代表我们不用这次交易名额

if(k){    //代表我们使用这次的交易名额

for (i = 1;i<=n-1;i++)    //枚举和前面的某一天进行交易

dp[n][k] = max(dp[i][k-1] + arr[i] - arr[n]) 

}

复杂度O(n^2*k)

关键是有个for循环在这里,我们可不可以把它优化掉呢?假如想优化掉这个for循环的话,我们必须得出每天之间的关系。天与天之间的关系很难用盈余来表示,但是我们可以用盈亏来表示,所以我们这里需要把状态设置为: dp[n][k][...],代表第n天我们还有k次交易机会的盈亏值,注意因为有亏值在这里,所以这一个值有可能为负数。后面的...表示,状态还需要有一点补充,经过再次观察,我们得到其实每天只有两种状态,只有股票和不持有股票。所以我们再详细一点,dp状态设计为dp[n][k][sta],其中sta = 0 表示这一天我们不持有股票, sta = 1 表示这一天我们继续持有股票,或者购买新股票!

那么我们就可以得到:

dp[n][k][sta] = max(dp[n-1][k][0],dp[n-1][k][1] + arr[n]) if sta == 0    //代表这一天我们是继续不持有股票,还是卖掉股票

dp[n][k][sta] = max(dp[n-1][k][1],dp[n-1][k-1][0] - arr[n]) if sta == 1 //代表我们这一天是继续持有股票或者是买股票
const int inf = 1e9;
const int FLAG  = 1e9;
class Solution {
public:
vector<int> arrmv;
    vector<vector<vector<int>>> dp;
    int dfs(int n,int k,int sta){
        if(n == 0)return (sta==1)?-inf:0;
        if(dp[n][k][sta] != FLAG)return dp[n][k][sta];
        if(sta == 0)
        dp[n][k][sta] = max(dfs(n-1,k,0),dfs(n-1,k,1) + arrmv[n-1]);
        else {
            dp[n][k][sta] = dfs(n-1,k,1);
            if(k)dp[n][k][sta] = max(dp[n][k][sta],dfs(n-1,k-1,0) - arrmv[n-1]);
        }
        return dp[n][k][sta];
    }
    int maxProfit(int k,vector<int>& prices) {
        arrmv = prices;
        int n = arrmv.size();
        if(k>n/2) {    //特殊情况,需要优化一下.
            int ret = 0;
            for(int i = 1;i<n;i++)ret+=(arrmv[i]>arrmv[i-1])?arrmv[i]-arrmv[i-1]:0;
            return ret;
        }
        dp.assign(n+10,vector<vector<int>>(k+10,vector<int>(2,FLAG)));
        return dfs(n,k,0);
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值