【华为机试】2023年真题B卷（python）-最大股票收益

### 华为OD机试分糖果问题Python3解法 #### 题目描述 在一个班级里有 `n` 名学生，每位学生都有一个评分。老师有一些糖果要分配给这些学生，满足以下条件： - 每位学生至少得到一颗糖果。 - 如果一位学生的评分严格高于旁边的学生，则该学生应获得更多的糖果。 编写函数来计算并返回老师至少需要准备多少颗糖果才能满足上述两个条件[^1]。 #### Python3 实现方案 为了实现这一目标，可以通过两次遍历的方法解决这个问题。第一次从前向后遍历数组，第二次从后向前遍历数组，在这两个过程中分别记录每个位置上应该有的最小糖果数，最后取两者中的最大值作为最终的结果。 ```python def candy(ratings): n = len(ratings) candies = [1] * n # Forward pass to ensure higher rated students get more candies than the previous student. for i in range(1, n): if ratings[i] > ratings[i - 1]: candies[i] = candies[i - 1] + 1 # Backward pass to ensure higher rated students also get more candies than next student. for i in range(n - 2, -1, -1): if ratings[i] > ratings[i + 1]: candies[i] = max(candies[i], candies[i + 1] + 1) return sum(candies) ``` 此算法的时间复杂度为 O(N)，其中 N 是输入列表的长度；空间复杂度同样为 O(N)，因为我们创建了一个额外的列表用于存储每名同学对应的最少所需糖果数量。