Leetcode309. 最佳买卖股票时机含冷冻期(C语言)
算法-动态规划(股票交易):算法与数据结构参考
题目:
给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。计算出最大利润。
尽可能地完成更多的交易(多次买卖一支股票);
你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。例:
输入: [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
思路:
动态规划。
假定dp[i][j]为第i天结束时的“j”状态;
j=0,表示不持有状态,前一天不持有/当天卖出
j=1,表示持有状态,前一天持有/买入
j=2,表示冻结,前一天卖出
初始化,第一天结束时状态:
dp[0][0]=0 (第一天不买入)
dp[0][1]=-prices[0] (第一天买入)
代码:
#define max(a,b) (a>b)?a:b
int maxProfit(int* prices, int pricesSize){
if(!prices || pricesSize==0) return 0;
int dp[pricesSize][3];
dp[0][0]=0;dp[0][1]=-prices[0];dp[0][2]=0; //第一天结束,0卖出,1买入,2冻结
for(int i=1;i<pricesSize;i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][2]); //当天结束无股,前一天无股/保持
dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]); //当天结束有股,前一天持股/买入
dp[i][2]=dp[i-1][1]+prices[i]; //冻结,前一天卖出
}
return max(dp[pricesSize-1][0], dp[pricesSize-1][2]); //保持无股/卖出冻结
}