leetcode 714. 買賣股票的最佳時機含手續費 best-time-to-buy-and-sell-stock-with-transaction-fee
阿新 • • 發佈:2018-12-12
給定一個整數陣列 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]; } };