题意
假设把某股票的价格按照时间先后顺序存储在数组中,请问买卖该股票一次可能获得的最大利润是多少?
示例 1:
输入: [7,1,5,3,6,4]
输出: 5
解释: 在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。
限制:
0 <= 数组长度 <= 10^5
问题分析
蛮力法的时间复杂度为 O ( n 2 ) O(n^2) O(n2)
应该用动态规划的思想来解决,设dp[i]代表以prices[i]为结尾的子数组的最大利润,因此 dp[i] 等于 dp[i-1] 和第 i 日卖出的最大利润中的最大值。
d
p
[
i
]
=
m
a
x
(
d
p
[
i
−
1
]
,
p
r
i
c
e
s
[
i
]
−
m
i
n
(
p
r
i
c
e
s
[
0
:
i
]
)
)
dp[i] = max(dp[i-1], prices[i] - min(prices[0:i]))
dp[i]=max(dp[i−1],prices[i]−min(prices[0:i]))
在遍历prices时可以用一个变量min保存每日最低价格,而又因为 dp[i] 只跟 dp[i-1],prices[i],min相关,因此可使用一个变量 maxDiff 代替dp列表,优化的方程为:
m
a
x
D
i
f
f
=
m
a
x
(
m
a
x
D
i
f
f
,
p
r
i
c
e
s
[
i
]
−
m
i
n
(
c
o
s
t
,
p
r
i
c
e
s
[
i
]
)
maxDiff = max(maxDiff, prices[i] - min(cost, prices[i])
maxDiff=max(maxDiff,prices[i]−min(cost,prices[i])
class Solution {
public:
int maxProfit(vector<int>& prices) {
int cost = INT_MAX;
int maxDiff = 0;
for(int price : prices){
cost = min(cost, price);
maxDiff = max(maxDiff, price-cost);
}
return maxDiff;
}
};