代码随想录第四十九天
121. 买卖股票的最佳时机
给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0 。
在本题中股票只能交易一次,因此在假设股票持有前现金总和是0,则在动规数组dp中dp[i][0]表示第i天持有股票的现金做多现金,dp[i][1]表示在第i天未持有(即股票在第i天之前已经被卖出,或在第i天即将卖出)时手里获得的最大现金额度
则递推公式为
dp[i][0]=max(dp[i-1][0],0-price[i])
dp[i][1]=max(dp[i-1][1],dp[i-1][0]+price[i])
遍历从前到后
初始化为dp[0][0]=0-price[0] dp[0][1]=0;
模拟动态规划过程
代码如下
int sellStockOnce(vector<int>& price)
{
vector<vector<int>>dp(price.size(), vector<int>(2, 0));
dp[0][0] = -price[0];
for (int i=1;i<price.size();i++)
{
dp[i][0] = max(dp[i - 1][0], -price[i]);
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + price[i]);
}
return dp[price.size()-1][1];
}
122.买卖股票的最佳时机II
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
本题上一题类似,不过改成了股票可以多次交易,那么在每次股票持有前,现金额变成了dp[i][1]
动规数组dp中dp[i][0]表示第i天持有股票的现金做多现金,dp[i][1]表示在第i天未持有(即股票在第i天之前已经被卖出,或在第i天即将卖出)时手里获得的最大现金额度
则递推公式为
dp[i][0]=max(dp[i-1][0],dp[i-1][1]-price[i])
dp[i][1]=max(dp[i-1][1],dp[i-1][0]+price[i])
遍历从前到后
初始化为dp[0][0]=0-price[0] dp[0][1]=0;
模拟动态规划过程
代码如下
int sellStockMultiple(vector<int>& prices)
{
vector<vector<int>>dp(prices.size(), vector<int>(2, 0));
dp[0][0] = -prices[0];
for (int i=1;i<prices.size();i++)
{
dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i]);
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
}
return dp[prices.size() - 1][1];
}