题目
给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入:k = 2, prices = [2,4,1] 输出:2 解释:在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。
示例 2:
输入:k = 2, prices = [3,2,6,5,0,3] 输出:7 解释:在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。 随后,在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。
解题
方法一:动态规划
思路和leetcode-123:买卖股票的最佳时机 III是一样的。只不过之前那个是限制买卖2次,这个是k次。
dp[i][1],dp[i][3],....
代表买入
dp[i][2],dp[i][4],....
代表卖出
class Solution { public: int maxProfit(int k, vector<int>& prices) { int n=prices.size(); if(n==0) return 0; vector<vector<int>> dp(n,vector<int>(2*k+1)); for(int i=1;i<2*k;i+=2){ dp[0][i]=-prices[0]; } for(int i=1;i<n;i++){ dp[i][0]=dp[i-1][0]; for(int j=1;j<2*k;j+=2){ dp[i][j]=max(dp[i-1][j],dp[i-1][j-1]-prices[i]); dp[i][j+1]=max(dp[i-1][j+1],dp[i-1][j+1-1]+prices[i]); } } return dp[n-1][2*k]; } };