题目:假设你有一个数组,其中第 i 个元素是股票在第 i 天的价格。
你有一次买入和卖出的机会。(只有买入了股票以后才能卖出)。请你设计一个算法来计算可以获得的最大收益。
这题一看刚开始有个理解误区,想着找出最大的index和最小的index,保证最最小的index小于最大的index即可,然后求最大利润。但是跑用例的时候却发现,这个最大和最小都是相对的,没办法从整个变化的区间中进行确定,所以还是需要用动规来做。
对于动规一直比较迷,似会非会的感觉。
首先确定动规方程,对于第i天来说,可以卖出获得 dp[i]-前i天最小的价格的利润,也可以不卖保留dp[i-1]的利润,然后第i天的利润应该取两种选择的最大值。
首先确定dp[0]=0,然后定义一个cost参数来动态更新前i天最小的购买价格,对于每一天,取两种情况的max作为dp[i],最终返回dp[dp.length-1]即可。
完整代码如下:
public int maxProfit (int[] prices) {
int [] dp = new int[prices.length];
dp[0] = 0;
int cost = prices[0];
for(int i=1;i<prices.length;i++) {
cost = Math.min(cost, prices[i]);
dp[i] = Math.max(dp[i-1], prices[i]-cost);
}
return dp[prices.length-1];
}