LintCode--best-time-to-buy-and-sell-stock-ii(买卖股票的最佳时机 II)
原题链接:http://www.lintcode.com/zh-cn/problem/best-time-to-buy-and-sell-stock-ii/
假设有一个数组,它的第i个元素是一个给定的股票在第i天的价格。设计一个算法来找到最大的利润。你可以完成尽可能多的交易(多次买卖股票)。然而,你不能同时参与多个交易(你必须在再次购买前出售股票)。
样例
给出一个数组样例[2,1,2,0,1], 返回 2
分析:
只要收益为正的,就加上。
时间复杂度 O(n)
代码(C++、Python、Java)
class Solution {
public:
/**
* @param prices: Given an integer array
* @return: Maximum profit
*/
int maxProfit(vector<int> &prices) {
// write your code here
int n = prices.size();
int res = 0;
for (int i = 0; i < n-1; i ++)
if (prices[i+1] - prices[i] > 0)
res += prices[i+1] - prices[i];
return res;
}
};
class Solution {
/**
* @param prices: Given an integer array
* @return: Maximum profit
*/
public int maxProfit(int[] prices) {
// write your code here
int n = prices.length;
int res = 0;
for (int i = 0; i < n-1; i++)
if (prices[i+1] - prices[i] > 0)
res += prices[i+1] - prices[i];
return res;
}
};
class Solution:
"""
@param prices: Given an integer array
@return: Maximum profit
"""
def maxProfit(self, prices):
# write your code here
n = len(prices)
res = 0
for i in range(n-1):
if prices[i+1] - prices[i] >= 0:
res += prices[i+1] - prices[i]
return res