又是一道买卖股票的题目,题意和之前的差不多。不过又有了新的条件:不限制买卖的次数,但刚卖出股票的第二天不能买入股票。
这道题用动态规划很快就能解决,令d[j] 为区间[0, j] 的最大利润递推公式如下:
当j = 0 时, d[j] = 0 ;
当1 <= j <= n-1 时,d[j] = max(d[j-1], prices[j] - prices[0], prices[j] - prices[1], d[i] + prices[j] - prices[i+2] ) ;其中 0 <= i <= j - 3 ;
公式的正确性不好说明,但仔细思考后发现这样的递推公式确实不会丢失正确的解。代码如下,复杂度为O(n2):
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size() ;
if (n == 0 || n == 1) return 0 ;
vector<int> d ;
for (int i=0; i<n; i++) d.push_back(0) ;
for (int j=1; j<n; j++) {
d[j] = max(d[j-1], prices[j] - prices[0]) ;
d[j] = max(d[j], prices[j] - prices[1]) ;
for (int i=0; i<j-2; i++) {
d[j] = max(d[j], d[i] + prices[j] - prices[i+2]) ;
}
// cout << d[j] << endl ;
}
return d[n-1] ;
}
};
代码很简单。值得一提的是这道题有内存限制,曾经开过一个二维数组a[i][j] 来存储[i,j]之间至多买卖一次的最大利润结果超出内存空间了。
LeetCode上股票相关的题目终于做完了...