2. 换水问题

给你两个整数 numBottles 和 numExchange 。

numBottles 代表你最初拥有的满水瓶数量。在一次操作中，你可以执行以下操作之一：

• 喝掉任意数量的满水瓶，使它们变成空水瓶。
• 用 numExchange 个空水瓶交换一个满水瓶。然后，将 numExchange 的值增加 1 。

注意，你不能使用相同的 numExchange 值交换多批空水瓶。例如，如果 numBottles == 3 并且 numExchange == 1 ，则不能用 3 个空水瓶交换成 3 个满水瓶。

返回你 最多 可以喝到多少瓶水。

示例 1：

输入：numBottles = 13, numExchange = 6
输出：15
解释：上表显示了满水瓶的数量、空水瓶的数量、numExchange 的值，以及累计喝掉的水瓶数量。

示例 2：

输入：numBottles = 10, numExchange = 3
输出：13
解释：上表显示了满水瓶的数量、空水瓶的数量、numExchange 的值，以及累计喝掉的水瓶数量。

提示：

• 1 <= numBottles <= 100
• 1 <= numExchange <= 100

class Solution:def maxBottlesDrunk(self, numBottles: int, numExchange: int) -> int:total = numBottleswhile numBottles > 0:numBottles = numBottles - numExchangeif numBottles >= 0:total += 1else:breaknumExchange += 1numBottles += 1return total

3. 交替子数组计数

给你一个二进制数组nums 。

如果一个子数组中 不存在 两个 相邻 元素的值 相同 的情况，我们称这样的子数组为 交替子数组 。

返回数组 nums 中交替子数组的数量。

示例 1：

输入： nums = [0,1,1,1]

输出： 5

解释：

以下子数组是交替子数组：[0] 、[1] 、[1] 、[1] 以及 [0,1] 。

示例 2：

输入： nums = [1,0,1,0]

输出： 10

解释：

数组的每个子数组都是交替子数组。可以统计在内的子数组共有 10 个。

提示：

• 1 <= nums.length <= 105
• nums[i] 不是 0 就是 1 。

class Solution:def countAlternatingSubarrays(self, nums: List[int]) -> int:ans = cnt = 0for i, x in enumerate(nums):if i and x == nums[i - 1]:cnt = 1else:cnt += 1ans += cnt  # 有 cnt 个以 i 为右端点的交替子数组return ans