1. 程式人生 > >leetcode 714. 買賣股票的最佳時機含手續費 best-time-to-buy-and-sell-stock-with-transaction-fee

leetcode 714. 買賣股票的最佳時機含手續費 best-time-to-buy-and-sell-stock-with-transaction-fee

給定一個整數陣列 prices,其中第 i 個元素代表了第 i 天的股票價格 ;非負整數 fee 代表了交易股票的手續費用。

你可以無限次地完成交易,但是你每次交易都需要付手續費。如果你已經購買了一個股票,在賣出它之前你就不能再繼續購買股票了。

返回獲得利潤的最大值。

示例 1:

輸入: prices = [1, 3, 2, 8, 4, 9], fee = 2
輸出: 8
解釋: 能夠達到的最大利潤:  
在此處買入 prices[0] = 1
在此處賣出 prices[3] = 8
在此處買入 prices[4] = 4
在此處賣出 prices[5] = 9
總利潤: ((8 - 1) - 2) + ((9 - 4) - 2) = 8.

注意:

  • 0 < prices.length <= 50000.
  • 0 < prices[i] < 50000.
  • 0 <= fee < 50000.

解法:

使用dp[i][0]代表第i天手中沒有股票時能賺的最高價,dp[i][1]代表第i天手中有股票時所賺的最高價

另外假設購買股票不用給手續費,在賣出的時候給,這樣會簡化不少特判

可以得到轉移方程:

dp[i][0]=max(dp[i-1][1]+prices[i]-fee,dp[i-1][0]);
dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1]);

這兩個轉移方程的意思是:當第i天手裡沒有股票時,這一天有兩個可能,可能是在今天賣出了股票,從而一共賺取了dp[i-1][1]+prices[i]-fee,也有可能是昨天就沒有股票,今天只是仍然沒有股票而已,這個時候dp[i][0]=dp[i-1][0]

同理可以推匯出dp[i][1]的轉移方程

最終答案就是dp[n][0],因為必然要讓手裡的錢更多,肯定是手裡沒有股票的時候

程式碼:

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