题目: Best Time to Buy and Sell Stock II
假设有一个数组,它的第 i 个元素是一个给定的股票在第 i 天的价格。
设计一个算法来找到最大的利润。你可以完成尽可能多的交易(多次买卖股票)。然而,你不能同时参与多个交易(你必须在再次购买前出售股票)。
思路:
因为是找最大的利润,只要后一天的价格大于今天的价格,就可以做差价计算到总计润中去。
最终代码:
public class MaxProfile {
@Test
public void test() {
Assert.assertEquals(0, maxProfit(new int[]{}));
Assert.assertEquals(0, maxProfit(new int[]{10, 1}));
Assert.assertEquals(9, maxProfit(new int[]{1, 10}));
Assert.assertEquals(9, maxProfit(new int[]{1, 10, 10}));
Assert.assertEquals(9, maxProfit(new int[]{1, 10, 10, 10, 10}));
Assert.assertEquals(9, maxProfit(new int[]{1, 1, 10, 10}));
Assert.assertEquals(18, maxProfit(new int[]{1, 1, 10, 10, 1, 10}));
Assert.assertEquals(9, maxProfit(new int[]{1, 1, 10, 10, 1, 1, 1, 0}));
Assert.assertEquals(17, maxProfit(new int[]{10, 0, 1, 10, 3, 5, 6, 5, 6, 4, 7, 1}));
}
public int maxProfit(int[] prices) {
if (prices == null || prices.length < 2) {
return 0;
}
int sum = 0;
for (int i = 0; i < prices.length - 1; i++) {
if (prices[i + 1] > prices[i]) {
sum += prices[i + 1] - prices[i];
}
}
return sum;
}
}