- 算法
动态规划 - 核心思想
1)做了几个动态方程的题,觉得第一步是先找当前有哪几种状态。在本题中,共分为三种状态:
(1)不持有:有两种情况导致不持有①i-1天不持有股票;②i-1天有股票且当天出货
(2)持有:①昨天持有股票;②今天买入前提是昨天没出货
(3)不持有因为当天出货,此状态两个目的:满足(1)不持有状态下第二种情况;确保昨天没有出货。
2)接着确定初始状态:
dp[0][0] = 0;
dp[0][1] = -1 * prices[0];
dp[0][2] = 0;
3)最后需要确保结果有可能是当天出货。
- 代码
class Solution {
public int maxProfit(int[] prices) {
if(prices.length <= 1) return 0;
int[][] dp = new int[prices.length][3];
dp[0][0] = 0;
dp[0][1] = -1 * prices[0];
dp[0][2] = 0;
for (int i = 1; i < prices.length; ++i) {
dp[i][0] = Math.max(dp[i-1][0],dp[i-1][2]);
dp[i][1] = Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
dp[i][2] = dp[i-1][1] + prices[i];
}
return Math.max(dp[prices.length-1][0],dp[prices.length-1][2]);
}
}