题目
给定一个数组,它的第 i 个元素是一支给定股票第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易(多次买卖一支股票)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
我的想法
我在读题之后就理解错题意了。题意是“可以在同一天买入或卖出”,我理解错了。
看来题解后打算使用波峰波谷方法。
代码
class Solution {
public:
int maxProfit(vector<int>& prices) {
int profit = 0;
int memory[2] = {-1, -1};
for(int i = 0;i<prices.size();i++){
if (memory[0]==-1){
memory[0] = prices[i];
}
else if (memory[1]==-1){
if (prices[i]>memory[0]){
memory[1] = prices[i];
}
else {
memory[0] = prices[i];
}
}
else {
if (prices[i]<memory[1]){
profit += memory[1] - memory[0];
memory[0] = prices[i];
memory[1] = -1;
}
else {
memory[1] = prices[i];
}
}
}
if (memory[1]!= -1){
profit += memory[1] - memory[0];
}
return profit;
}
};
上面的代码能通过,但是效果不太好。
题解的代码更加巧妙了,只要明天比今天高,就是利润。
class Solution {
public:
int maxProfit(vector<int>& prices) {
int profit = 0;
for(int i = 0;i<prices.size()-1;i++){
if (prices[i]<prices[i+1]){
profit += (prices[i+1] - prices[i]);
}
}
return profit;
}
};