题目
//来自leetcode
给定一个整数数组,其中第 i 个元素代表了第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
输入示例
1 2 3 0 2
输出示例
3
解题思路
这道题也是北航2017级算法考试的一道中等题,考点是动态规划。考场上想到了正确的状态专业方程,但没敢尝试做题。
动态转移方程如下,我们维护买入和卖出两个数组,卖出后,必须隔一天才能买入,所以有了buy数组的转移方程。这个思路其实也可以解决没有冷冻期的股票买卖问题,只要把sell[i-2]变为sell[i-1]即可。不得不说,动态规划,还是很玄妙很有意思的东西。
buy[i]=max(sell[i-2]-price[i],buy[i-1])
sell[i]=max(buy[i-1]+price[i],sell[i-1])
leetcodeAC代码
//该代码在leetcode英文站提交通过,但中文站显示为待审,并超时,不知道原因何在
//因北航oj不开放考试题目,所以也就不提供带输入输出的完整代码了(避免坑到其他人)
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n=prices.size();
if(n==0||n==1)
return 0;
int buy[n],sell[n];
buy[0]=-prices[0];
sell[0]=0;
buy[1]=max(-prices[0],-prices[1]);
sell[1]=max(0,buy[0]+prices[1]);
for(int i=2;i<n;i++)
{
buy[i]=max(sell[i-2]-prices[i],buy[i-1]);
sell[i]=max(sell[i-1],buy[i-1]+prices[i]);
}
return sell[n-1];
}
};