华为OD机试 - 最多购买宝石数目 (python 2024年C卷D卷)

华为OD机试（C卷+D卷）2024真题目录(Java & c++ & python)

题目描述

橱窗里有一排宝石，不同的宝石对应不同的价格，宝石的价格标记为 gems[i]
0 ≤ i < n
n = gems.length
宝石可同时出售0个或多个，如果同时出售多个，则要求出售的宝石编号连续；

例如客户最大购买宝石个数为m，购买的宝石编号必须为：gems[i]，gems[i+1]，…，gems[i+m-1]
0 ≤ i < n
m ≤ n
假设你当前拥有总面值为 value 的钱，请问最多能购买到多少个宝石，如无法购买宝石，则返回0。

输入描述

第一行输入n，参数类型为int，取值范围：[0,10^6]，表示橱窗中宝石的总数量。

之后 n 行分别表示从第0个到第n-1个宝石的价格，即 gems[0] 到 gems[n-1] 的价格，类型为int，取值范围：(0,1000]。

之后一行输入v，类型为int，取值范围：[0,10^9]，表示你拥有的钱。

输出描述

输出int类型的返回值，表示最大可购买的宝石数量。

示例1

输入

7
8
4
6
3
1
6
7
10

输出

3

说明

gems = [8,4,6,3,1,6,7], value = 10

最多购买的宝石为gems[2]至gems[4]或者gems[3]至gems[5]

示例2

输入

0
1

输出

0

说明

gems = [], value = 1 因为没有宝石，所以返回0

示例3

输入

9
6
1
3
1
8
9
3
2
4
15

输出

4

说明

gems = [6, 1, 3, 1, 8, 9, 3, 2, 4], value = 15 最多购买的宝石为gems[0]至gems[3]

示例4

输入

9
1
1
1
1
1
1
1
1
1
10

输出

9

说明

gems = [1, 1, 1, 1, 1, 1, 1, 1, 1], value = 10
最多购买的宝石为gems[0]至gems[8]，即全部购买

解题思路

双指针/滑动窗口。

以用例1为例：蓝色为L指针，橙色为R指针


其中需要注意的是，如果内部和超过了v，那么此时我们需要不停右移L指针，来减少内部和，直到内部和<=v时，结束L指针右移。然后才能开始R指针右移。

还有一种特殊情况比如：

gems = [1,2,3,100,4,5,6]，v = 10

参考代码

# 输入获取
n = int(input())  # 橱窗中宝石的总数量
gems = [int(input()) for _ in range(n)]  # n个宝石的价格
v = int(input())  # 你拥有的钱


# 记录题解
ans = 0

# 滑动窗口左右边界
l = 0
r = 0

# 滑动窗口和
window_sum = 0

while r < n:
    if window_sum + gems[r] <= v:
        window_sum += gems[r]
        r += 1
    else:
        ans = max(ans, r - l)
        window_sum -= gems[l]
        l += 1

# 收尾
ans = max(ans, r - l)  # 无需再判断window_sum，因为循环结束时已确保

return ans