Leetcode 题解 [121.买卖股票的最佳时机](C++)
Leetcode 题解(每日打卡)
[121.买卖股票的最佳时机]
题目描述
给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0
思路
利用一个大小为4的数组,分别记录min、min_index、max、max_index
示例1理解:
原: [7,1,5,3,6,4]
最大值:[0,0,4,2,5,3]
示例2理解:
原: [7,6,4,3,1]
最大值:[0,0,0,0,0]
这里我使用了一个max变量来替代上述的最大值数组
这里记录min和max下标的意义在与便于复位,就有点类似于哨兵一样,防止max的index小于min
代码实现
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n[4];//0:min,1:mind_index,2:max,3:max_index
n[0] = prices[0];
n[2] = prices[0];
n[1] = 0;
n[3] = 0;
int max = -1;
for(int i=0;i<prices.size();i++){
if(n[0]>prices[i]){
n[0] = prices[i];
n[1] = i;
}
if(n[2]<prices[i]){
n[2] = prices[i];
n[3] = i;
}
if(n[3] < n[1]){
n[3] = n[1];
n[2] = n[0];
}
if(n[2] - n[0] > max)
max = n[2]-n[0];
}
return max;
}
};
时间复杂度: O(n)
空间复杂度: O(1)