题目: 算法
给定一个整数数组,其中第 i 个元素表明了第 i 天的股票价格 。设计一个算法计算出最大利润。在知足如下约束条件下,你能够尽量地完成更多的交易(屡次买卖一支股票):数组
a.你不能同时参与多笔交易(你必须在再次购买前出售掉以前的股票)。
b.卖出股票后,你没法在次日买入股票 (即冷冻期为 1 天)。spa
题解:设计
此题是典型的动态规划解法。动态规划2步走:code
1.状态定义blog
对于股票咱们分为持有、不持有2种状态: it
memo[i][0]表示第i天不持有股票的最大利润
memo[i][1]表示第i天持有股票的最大的利润
2.状态转移方程 io
//第i天不持有股票的最大利润 = Math.max(昨天就不持有,昨天持有且今天卖出)
memo[i][0] = Math.max(memo[i-1][0],memo[i-1][1]+prices[i]);function
//第i天持有股票的最大的利润 = Math.max(昨天就持有,前天不持有且今天买入),i-2表示前天,要作越界判断
memo[i][1] = Math.max(memo[i-1][1],memo[i-2>0?i-2:0][0] - prices[i]);class
下面上代码:
Java版本
public int maxProfit(int[] prices) { if(prices == null || prices.length <=1) return 0; //记忆数组memo int[][] memo = new int[prices.length][2]; //memo[i][0]表示第i天不持有股票的最大利润,memo[i][1]表示第i天持有股票的最大的利润 memo[0][0] = 0; memo[0][1] = - prices[0];//第一天持有股票的利润 for(int i=1;i<prices.length;i++){ memo[i][0] = Math.max(memo[i-1][0],memo[i-1][1] + prices[i]); memo[i][1] = Math.max(memo[i-1][1],memo[i-2>0?i-2:0][0] - prices[i]); } //最后一天不持有股票时的利润最大 return memo[prices.length-1][0]; }
JS版本
var maxProfit = function(prices) { if(prices == null || prices.length <= 1) return 0; //记忆数组 let memo = []; for(let i=0;i<prices.length;i++){ memo.push([]); } //memo[i][0]表示第i天不持有股票的最大利润,memo[i][1]表示第i天持有股票的最大的利润 memo[0][0] = 0; memo[0][1] = - prices[0]; for(let i=1;i<prices.length;i++){ //第i天不持有股票的最大利润 = Math.max(昨天就不持有,昨天持有且今天卖出) memo[i][0] = Math.max(memo[i-1][0],memo[i-1][1]+prices[i]); //第i天持有股票的最大的利润 = Math.max(昨天就持有,前天不持有且今天买入),i-2表示前天,要作越界判断 memo[i][1] = Math.max(memo[i-1][1],memo[i-2>0?i-2:0][0] - prices[i]); } return memo[prices.length-1][0]; };