股票投资问题
这是要一个系列的动态规划问题
dp[i][k][j]:其中i代表第几天【0-N】,k代表可以买卖股票的次数[0-K],j代表当天是否持有股票[0,1] dp为最大的利润
因为动态规划肯定是转移方程,所以我们有
dp[i][k][0]=max{dp[i-1][k][0],dp[i-1][k][1]+prices[i]}
dp[i][k][1]=max{dp[i-1][k][1],dp[i-1][k][0]-prices[i]}
再看看基础情况
1:如果没有买卖次数,肯定值为0:dp[i][0][j]=0
2:在第1天【I=0】的时候:
如果选择购买股票,则,
dp[0][k][1]=-prices[i]
如果选择不购买股票,则
dp[0][k][0]=0
接下来就解答题目:
题目一:k=1
dp[i][1][0]=max{dp[i-1][1][0],dp[i-1][1][1]+prices[i]}
dp[i][1][1]=max{dp[i-1][1][1],dp[i-1][0][0]-prices[i]}
因为dp[i-1][0][0]=0,又因为上述式中k全为1,可简化为:
dp[i][0] = Math.max(dp[i-1][0], dp[i-1][1] + prices[i]);
dp[i][1] = Math.max(dp[i-1][1], -prices[i]);
得到答案:
class Solution {
public int maxProfit(int[] prices) {
int [][]dp=new int [prices.length][2];
if(prices.length==0)return 0;
dp[0][0]=0;
dp[0][1]=-prices[0];
for(int i=1;i<prices.length;i++){
dp[i][0] = Math.max(dp[i-1][0], dp[i-1][1] + prices[i]);
dp[i