【华为OD机试2023】贪心的商人(Python)

题目描述：

商人经营一家店铺，有number种商品，由于仓库限制每件商品的最大持有数量是item[index]，每种商品的价格在每天是item_price[item_index][day]，通过对商品的买进和卖出获取利润，请给出商人在days天内能获取到的最大的利润；

注： 同一件商品可以反复买进和卖出；

输入描述：

3  // 输入商品的数量 number

3  // 输入商人售货天数 days

4 5 6  // 输入仓库限制每件商品的最大持有数量是item[index]

1 2 3   // 输入第一件商品每天的价格

4 3 2   // 输入第二件商品每天的价格

1 5 3   // 输入第三件商品每天的价格

输出描述：

32 // 输出商人在这段时间内的最大利润

补充说明：

根据输入的信息：

number = 3

days = 3

item[3] = {4，5，6}

item_price[3][4] = {{1，2，3}，{4，3，2}，{1，5，3}}

针对第一件商品，商人在第一天的价格是item_price[0][0] = 1时买入item[0]件，在第三天item_price[0][2] = 3的时候卖出，获利最大是8；

针对第二件商品，不进行交易，获利最大是0；

针对第三件商品，商人在第一天的价格是item_price[2][0] = 1时买入item[2]件，在第二天item_price[2][0] == 5的时候卖出，获利最大是24；

因此这段时间商人能获取的最大利润是8 + 24 = 32；

示例1

输入：

3

3

4 5 6

1 2 3

4 3 2

1 5 3

输出：

32

示例2

输入：

1

1

1

1

输出：

0

解题思路：

两层for循环，第一层用于遍历商品，第二层用于遍历某商品每天的价格

由于是允许多次买卖的，不能采用最高价-最低价的思路。只要当天的价格比前一天的高，我们就在前一天买进，当天卖出，然后不断累加利润

number=int(input())
days=int(input())
store=list(map(int,input().split()))
prices=[list(map(int,input().split())) for i in range(number)]
profit=0
for i,good_price in enumerate(prices):
    pre_price=good_price[0]
    for day_price in good_price:
        if day_price>pre_price:
            profit+=store[i]*(day_price-pre_price)
        pre_price=day_price
print(profit)