leetcode-121-线性DP-买卖股票的最佳时机

发表于 | 分类于

题目

1.png

解法

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
// 动态规划
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.empty() || prices.size() == 1) return 0;
		
        // dp[i][0]=x,表示今天没有持有股票,当前最大利润为x
        // dp[i][1]=x,表示今天持有股票,当前最大利润为x
        vector<vector<int>> dp(prices.size(), vector<int>(2, 0));、
        dp[0][0] = 0;
        dp[0][1] = -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], - prices[i]);
        } 

        return dp[prices.size() - 1][0];
    }
};

// 动态规划，优化空间复杂度
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(prices.empty() || n == 1) return 0;

        int dp_i_0 = 0;
        int dp_i_1 = -prices[0];
        for(int i = 1; i < n; ++i){
            dp_i_0 = max(dp_i_0, dp_i_1 + prices[i]);
            dp_i_1 = max(dp_i_1, - prices[i]);
        } 

        return dp_i_0;
    }
};