Description
Say you have an array for which the ith element is the price of a given stock on day i.
Design an algorithm to find the maximum profit. You may complete as many transactions as you like (ie, buy one and sell one share of the stock multiple times). However, you may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).
代码
class Solution {
public:
int maxProfit(vector<int>& prices) {
int profit = 0;
for (int i = 0; i < (int)prices.size() - 1; i++) {
if (prices[i] < prices[i + 1])
profit += prices[i+1] - prices[i];
}
return profit;
}
};
分析
由于可以进行多次交易,所以最好多进行交易,而不是在价格较低的时候买入,等到价格较高的时候卖出,因为这样可能会在等待的中间时间里少赚一些钱。
这道题的贪心算法应该是今天买的股票,如果第二天卖出能赚钱,就立即卖出。即使后面卖出能赚更多的钱,第二天卖出也不会影响最终结果,因为可以第二天卖了后再买入,然后第三天再卖出,最终的结果是不会有改变的。
如果第二天的股票价格比今天低,那今天就不买入股票,而是到第二天才买入,这样子的投入较少,等卖出时就能获得更多的收益。
基于这种思想,每天要做的事情就是比较今日股票和明日股票的价格。如果明天股票价格更高,就今天买入,明天卖出;如果今天价格更高,就今天不买,明天再买。这样子比较下去,就能求出问题的解。
由于vector的size函数返回值是无符号类型的整数,不能直接进行减1操作,需要进行类型转换。