题目:
给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0
思路:
1. 首先想到用动态规划, 使用一个矩阵存储,dp[i][j]表示以i位置开始,j位置结束的最大利润,其中j位置必须作为卖出日期,这样保证遍历完所有结点
其关系为:dp[i][j] = max(dp[i][j-1] + price[j]-price[j-1], 0)
2. 优化上述算法的空间复杂度,可以将矩阵存储改为平面, 由于我们下一个结束位置的状态只和上一个有关,所以可以采用tmp来存储上一个结束位置的计算结果
3. 继续优化时间复杂度,由于结果只需输出最大利润,并未要求输出位置,所以我们可以只循环一遍,改其复杂度为O(n)
代码:
public int maxProfit(int[] prices) {
int tmp = 0;
int max = 0;
for (int i = 1; i < prices.length; i++) {
tmp = Math.max(tmp + (prices[i] - prices[i-1]), 0);
if (tmp > max) {
max = tmp;
}
}
return max;
}