股票最高获利系列–动态规划:
- 买卖股票的最佳时机
- 买卖股票的最佳时机 II
- 买卖股票的最佳时机 III
- 买卖股票的最佳时机 IV
- 最佳买卖股票时机含冷冻期
- 买卖股票的最佳时机含手续费
算法框架:
// A code block
int maxProfit(int k, vector<int>& prices) {
int len=prices.size();
int a[len][k+1][2];
memset(a,0,sizeof(a));
for(int i=0;i<len;i++){
for(int j=1;j<=k;j++){
if(i==0){
a[i][j][0]=0;
a[i][j][1]=-prices[i];
continue;
}
a[i][j][0]=max(a[i-1][j][0],a[i-1][j][1]+prices[i]); //状态转移方程
a[i][j][1]=max(a[i-1][j][1],a[i-1][j-1][0]-prices[i]);
}
}
return a[len-1][k][0];
}
用状态a[i][j][0]表示第i天最多交易j次并且手上没买股票时的最大收入。
用状态a[i][j][1表示手上有股票的最大利润。
对于系列题目,如增加一天的冷冻时间只需要简单修改状态转移函数如下
// A code block
a[i][0]=max(a[i-1][0],a[i-1][1]+prices[i]);
a[i][1]=max(a[i-1][1],a[i-2][0]-prices[i]);
起初对于dp矩阵初始化部分有一些疑惑
// A code block
if(i==0){
a[i][j][0]=0;
a[i][j][1]=-prices[i];
continue;
}
对于矩阵第二维的理解是本题的关键,j可积极理解成最多经过j次交易,而不是正好经过j次交易的最大利润。