[golang] 122. Best Time to Buy and Sell Stock II

// [url]https://leetcode.com/studyplan/top-interview-150/[/url]
 
package topInterview
 
// maxProfitII
// 
// 122. Best Time to Buy and Sell Stock II
// 
// You are given an integer array prices where prices[i] is the price of a given stock on the ith day.
// 
// On each day, you may decide to buy and/or sell the stock. You can only hold at most one share of the stock at any time.
// However, you can buy it then immediately sell it on the same day.
// 
// Find and return the maximum profit you can achieve.
// 
// Example 1:
// 
// Input: prices = [7,1,5,3,6,4]
// Output: 7
// Explanation: Buy on day 2 (price = 1) and sell on day 3 (price = 5), profit = 5-1 = 4.
// Then buy on day 4 (price = 3) and sell on day 5 (price = 6), profit = 6-3 = 3.
// Total profit is 4 + 3 = 7.
// Example 2:
// 
// Input: prices = [1,2,3,4,5]
// Output: 4
// Explanation: Buy on day 1 (price = 1) and sell on day 5 (price = 5), profit = 5-1 = 4.
// Total profit is 4.
// Example 3:
// 
// Input: prices = [7,6,4,3,1]
// Output: 0
// Explanation: There is no way to make a positive profit, so we never buy the stock to achieve the maximum profit of 0.
//  
// 
// Constraints:
// 
// 1 <= prices.length <= 3 * 104
// 0 <= prices[i] <= 104
// 
// 
func maxProfitII(prices []int) int {
 
    // Инициировать накопленную прибыль
    profit := 0
 
    // Инициировать индекс текущей цены
    index := 1
 
    // Инициировать цикл
    for {
 
        // Если индекс достиг длины слайса цен
        if index >= len(prices) {
 
            // Прервать цикл
            break
        }
 
        // Если текущая цена больше предыдущей
        if prices[index] > prices[index-1] {
 
            // Прибавить к накопленной прибыли разность между текущей ценой и предыдущей
            profit += prices[index] - prices[index-1]
        }
 
        // Увеличить индекс текущей цены на один
        index++
    }
    // Вернуть накопленную прибыль
    return profit
}

func TestMaxProfitII(t *testing.T) {
 
    data := []struct {
        prices   []int
        expected int
    }{
        {
            []int{7, 1, 5, 3, 6, 4},
            7,
        },
        {
            []int{1, 2, 3, 4, 5},
            4,
        },
        {
            []int{7, 6, 4, 3, 1},
            0,
        },
    }
 
    for i, datum := range data {
 
        result := maxProfitII(datum.prices)
 
        if result != datum.expected {
            t.Errorf("unexpected result for test index %d expected [%+v] got [%+v]", i, datum.expected, result)
        }
    }
}