您现在的位置是:首页 >学无止境 >LeetCode 121 买卖股票的最佳时机网站首页学无止境
LeetCode 121 买卖股票的最佳时机
题目: 给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0 。
示例 1:
输入:[7,1,5,3,6,4]
输出:5
解释:在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。
示例 2:
输入:prices = [7,6,4,3,1]
输出:0
解释:在这种情况下, 没有交易完成, 所以最大利润为 0。
思路:
dp[i][0]表示第i天持有股票的最大金额
dp[i][1]表示第i天不持有股票的最大金额
首先看dp[i][0]持有的最大金额
如果是在第i天前就持有了,那么最大金额应该是dp[i-1][0]
如果是在第i天才持有的,那么最大金额就是-prices[i]
dp[i][0] = max(dp[i - 1][0],-prices[i])
再看dp[i][1]不持有的最大金额
如果是第i天之前就不持有了,那么最大金额应该是dp[i-1][1]
如果是第i天才不持有的,那么最大金额就是dp[i-1][0]+prices[i]
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i])
初始化:从递推公式可以看出来,和前一天的状态有关系
因此初始化前一天的就行,也就是第一天
持有:dp[0][0] = -prices[0]
不持有:dp[0][1] = 0
class Solution {
public:
int maxProfit(vector<int>& prices) {
if (prices.size() == 0)return 0;
vector<vector<int>> dp(prices.size(),vector<int>(2));
dp[0][0] = -prices[0];
dp[0][1] = 0;
for (int i = 1; i < prices.size();i++) {
dp[i][0] = max(dp[i - 1][0], -prices[i]);
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
}
return max(dp[prices.size()-1][0], dp[prices.size()-1][1]);
}
};
int main() {
vector<int> prices = { 7,1,5,3,6,4 };
Solution ss;
cout<<ss.maxProfit(prices)<<endl;
return 0;
}