LeetCode Best Time To Buy and Sell Stocks I II III IV

I

题意

要求：交易一次
（交易的定义为一次买入和一次卖出，并且进行下一次交易前上一次交易必须结束，且同一天不能同时进行买入和卖出的操作）

思路

求个前缀最小值和后缀最大值，枚举中间点
时间复杂度：$O(n)$

Code

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if (n == 0) return 0;
        int* minn = new int[n], *maxx=  new int[n];
        minn[0] = prices[0]; maxx[n-1] = prices[n-1];
        for (int i = 1; i < n; ++i) {
            minn[i] = min(minn[i-1], prices[i]);
        }
        for (int i = n-2; i >= 0; --i) {
            maxx[i] = max(maxx[i+1], prices[i]);
        }
        int ans = 0;
        for (int i = 0; i < n-1; ++i) {
            ans = max(ans, maxx[i+1]-minn[i]);
        }
        return ans;
    }
};

II

题意

要求：交易不限次数

思路

考虑一个单调增的序列，显然不会从中间交易，而是取头和尾。
所以就是把序列分为若干个尽量长的单调增序列，每个都取头和尾。
时间复杂度：$O(n)$

Code

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int i = 0, j, n = prices.size(), ans = 0;
        for (; i < n; i = j+1) {
            for (j = i; j < n-1; ++j) if (prices[j+1] <= prices[j]) break;
            ans += prices[j] - prices[i];
        }
        return ans;
    }
};

III

题意

要求：两次交易

思路

$dp[i][0]$ 表示第 $i$ 天及以前第一次买入，$dp[i][1]$ 表示第 $i$ 天及以前第一次卖出，$dp[i][2]$ 表示第 $i$ 天及以前第二次买入，$dp[i][3]$ 表示第 $i$ 天及以前第一次卖出。
转移方程十分显然（主要是要想到这么表示= =）

Code

class Solution {
    const int inf = 0x3f3f3f3f;
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if (n == 0) return 0;
        int dp[2][4];
        vector<int>& v = prices;
        dp[0][0] = -v[0], dp[0][1] = dp[0][2] = dp[0][3] = -inf;
        int ans = 0;
        for (int i = 1; i < n; ++i) {
            dp[i&1][0] = max(dp[(i-1)&1][0], -v[i]);
            dp[i&1][1] = max(dp[(i-1)&1][1], dp[(i-1)&1][0] + v[i]);
            dp[i&1][2] = max(dp[(i-1)&1][2], dp[(i-1)&1][1] - v[i]);
            dp[i&1][3] = max(dp[(i-1)&1][3], dp[(i-1)&1][2] + v[i]);
            ans = max(ans, max(dp[i&1][3], dp[i&1][1]));
        }
        return ans;
    }
};

IV

题意

要求：$k$ 次交易

思路

同上题

Code

class Solution {
    const int inf = 0x3f3f3f3f;
public:
    int maxProfit(int k, vector<int>& prices) {
        int n = prices.size();
        if (n == 0) return 0;
        k = min(n>>1, k);
        k <<= 1;
        int** dp = new int*[2];
        for (int i = 0; i < 2; ++i) dp[i] = new int[k];
        vector<int>& v = prices;
        dp[0][0] = -v[0];
        for (int i = 1; i < k; ++i) dp[0][i] = -inf;
        int ans = 0;
        for (int i = 1; i < n; ++i) {
            dp[i&1][0] = max(dp[(i-1)&1][0], -v[i]);
            for (int j = 1; j < k; ++j) {
                dp[i&1][j] = max(dp[(i-1)&1][j], dp[(i-1)&1][j-1]+((j & 1) ? v[i] : -v[i]));
                if (j&1) ans = max(ans, dp[i&1][j]);
            }
        }
        return ans;
    }
};