[动态规划] leetcode 714 Best Time to Buy and Sell Stock with Transaction Fee

problem:https://leetcode.com/problems/best-time-to-buy-and-sell-stock-with-transaction-fee/

        维护两个状态,一个是当前持有股票状态,一个是当前不持有股票状态,两者分别计算最大值。

        在第i天,如果当天不持有股票,那么可能是昨天也没有持有股票,也有可能是当天卖出了股票,所以才没有持有股票。我们从这两种状态中取最大值。

        在第i天,如果当前持有股票,那么可能是昨天就已经持有股票,也有可能是当天买了股票。我们从这两种状态中取最大值。

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int n = prices.size();
        vector<int> hasStock(n + 1, 0);
        vector<int> noStock(n + 1, 0);
        hasStock[0] = -50000;
        noStock[0] = 0;
        for(int i = 0;i < n;i++)
        {
            noStock[i + 1] = max(noStock[i], hasStock[i] + prices[i] - fee);
            hasStock[i + 1] = max(hasStock[i], noStock[i] - prices[i]);
        }
        return noStock[n];
    }
};

 

posted @ 2019-08-04 18:26  fish1996  阅读(195)  评论(0)    收藏  举报