题目
给你一个整数数组 prices
,其中 prices[i]
表示某支股票第 i
天的价格。
在每一天,你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买,然后在 同一天 出售。
返回 你能获得的 最大 利润 。
C++代码
#include <iostream>
#include <vector>
using namespace std;
/*
* 动态规划解决股票问题(自底向上)
* 根据买入卖出的规律确定状态转移方程
* 当第i天交易后没有股票可能是前一天没有股票,或者前一天有股票当天卖出
* 当第i天交易后有一只股票可能是前一天有股票,或者前一天没有当天买入
* 卖出的利息一定大于买入
*/
int maxPoint(vector<int>& prices) {
int n = prices.size();
int dp[20][2];
dp[0][0] = 0, dp[0][1] = -prices[0];
for (int i = 1; i < n; 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[n - 1][0];
}
int main() {
vector<int> prices = { 7, 1, 5, 3, 6, 4 };
cout << "The price: ";
for (int i = 0; i < prices.size(); i++) {
cout << prices[i] << " ";
}
cout << endl;
int dp = maxPoint(prices);
cout << "The max profit: " << dp << endl;
return 0;
}
分析
本题采用动态规划的方法,根据买入卖出的规律确定状态转移方程,当第 i 天交易后没有股票可能是前一天没有股票,或者前一天有股票当天卖出,当第 i 天交易后有一只股票可能是前一天有股票,或者前一天没有当天买入,卖出的利息一定大于买入。定义 dp[i][0] 为第 i 天交易后没有股票的利润,dp[i][1] 为第 i 天交易后有一股股票的利润,状态转移方程根据前一天是否有股票,是否卖出股票求出最大利润。
问题
这是第一次接触动态规划问题,比较难想出解法,一直靠解析的我推荐一篇文章更好地了解动态规划:https://zhuanlan.zhihu.com/p/365698607