题目大意:
给定一个数组,它的第 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);
}
};