最多购买宝石数目 - 华为OD统一考试-编程知识

最多购买宝石数目 - 华为OD统一考试

OD统一考试（C卷）

分值： 100分

题解： Java / Python / C++

alt

题目描述

橱窗里有一排宝石，不同的宝石对应不同的价格，宝石的价格标记为 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

示例2

输入：
0
1输出：
0说明：
因为没有宝石，所以返回 0

示例3

题解

滑动窗口的方法来解决。具体步骤如下：

初始化滑动窗口的左边界和右边界为0。
遍历右边界，累加宝石的价格，直到总价格大于等于拥有的钱或者遍历完所有宝石。
如果总价格小于等于拥有的钱，更新最大购买宝石数量。
移动左边界，减去左边界对应的宝石价格，直到总价格小于拥有的钱。
重复步骤2到步骤4，直到右边界遍历完所有宝石。

Java

import java.util.Scanner;/*** @author code5bug*/
public class Main {public static void main(String[] args) {Scanner scanner = new Scanner(System.in);// 输入宝石数量int n = scanner.nextInt();int[] gems = new int[n];// 输入宝石价格for (int i = 0; i < n; i++) gems[i] = scanner.nextInt();// 输入拥有的钱int v = scanner.nextInt();// 使用滑动窗口求解int left = 0, right = 0, total = 0, maxCount = 0;while (right < n) {total += gems[right];// 收缩左边界while (total > v) {total -= gems[left];left++;}// 更新最大购买宝石数量maxCount = Math.max(maxCount, right - left + 1);right++;}System.out.println(maxCount);}
}

Python

n = int(input())
gems = [int(input()) for _ in range(n)]
v = int(input())left, total, maxCount = 0, 0, 0
for right in range(n):total += gems[right]while total > v:total -= gems[left]left += 1maxCount = max(maxCount, right - left + 1)print(maxCount)

C++

#include <iostream>
#include <vector>using namespace std;int main() {int n;cin >> n;vector<int> gems(n);for (int i = 0; i < n; i++) {cin >> gems[i];}int v;cin >> v;int left = 0, right = 0, total = 0, maxCount = 0;while (right < n) {total += gems[right];// 收缩左边界while (total > v) {total -= gems[left];left++;}// 更新最大购买宝石数量maxCount = max(maxCount, right - left + 1);right++;}cout << maxCount << endl;return 0;
}