原题描述:
Say you have an array for which the ith element is the price of a given stock on day i.
If you were only permitted to complete at most one transaction (ie, buy one and sell one share of the stock), design an algorithm to find the maximum profit.
Example 1:
Input:[7,1,5,3,6,4]
Output:5
max. difference = 6 -1 =5
即:在一个数组中,第i
个元素代表的是第i
天的stock价格,要求在某天i
买入,在某天j
天卖出,使得收益最大。最多交易一次,可以不进行交易,即收益为0.
分析:
收益 = 售出时价格 - 买进时价格
如果现在已知结果是在i
天买入,在j
天卖出,这些天stock的价格是:
收益为:
aj−ai=(aj−aj−1)+(aj−1−aj−2)+...+(ai+1−ai)
,为从i
天到第j
天的收益之和,其中
(aj−aj−1),(aj−1−aj−2),...,(ai+1−ai)
都是相邻两天stock的差价difference,于是问题转换成:
prices数组转换成difference数组,求difference数组的最大子array和,即寻找最大子数列
最大子数列问题:
有数组 A=[a1,a2,...,an] ,从中寻找出一个连续的子数组 B=[ai,ai+1,...,aj−1,aj] ,使得 B 的和是所有子数组中最大的.
问题分析:
对于数组
- 如果
Bk+1
包含
k+1
以前的元素,那么
Bk+1=ak+1∪(含k的最大和)
,即
Bk+1=ak+1∪Bk
- 如果
Bk+1
不包含
k+1
以前的元素,那么
Bk+1=ak+1
综上所述,有Bk+1=max(ak+1,ak+1∪Bk)
此解法称为 Kadane’s Algorithm
问题要求的最大子序列和,可以转换成求
curmax=0
maxsofar=0
for i in range(len(A)):
curmax=max(A[i],A[i]+curmax)
maxsofar=max(curmax,maxsofar)
#maxsofar就是最大子序列和
回到本题售价中,则只需要将prices数组转换成差价数组之后,使用Kadane’s Algorithm即可解决!