解题思路:(python)
其实也不算是严格的二分法,就是分两次遍历。第一次遍历是从头至尾遍历,记录每个区间内一次股票交易获得最大利润,并保存至一个数组内,方便第二轮遍历取值,此处就是可以节省时间了。第二次遍历就是从尾向头遍历,选出相应坐标右边区间内一次交易的最大值,左边区间交易的最大值就可以直接从第一次遍历得到的数组获取,两者相加,即可得到两次交易的最大值。
class Solution(object):
def maxProfit(self, prices):
"""
:type prices: List[int]
:rtype: int
"""
min_price, total_max_profit = float('inf'), 0
# 记录顺序区间内每个坐标对应的利润最大值
first_profit = [0]*len(prices)
# 从左至右比遍历,获取顺序区间内一次交易利润的最大值
for i in range(len(prices)):
min_price = min(min_price, prices[i])
total_max_profit = max(total_max_profit, prices[i]-min_price)
first_profit[i] = total_max_profit
max_price = float('-inf')
max_profit = 0
'''
从右至左遍历
获取逆序区间内一次交易利润的最大值
两者相加得到两次利润的最大值
'''
for i in range(len(prices)-1, 0, -1):
max_price = max(max_price, prices[i])
max_profit = max(max_profit, max_price-prices[i])
total_max_profit = max(total_max_profit, max_profit+first_profit[i-1])
return total_max_profit