Leetcode 2389. 和有限的最长子序列

1.题目基本信息

1.1.题目描述

给你一个长度为 n 的整数数组 nums ，和一个长度为 m 的整数数组 queries 。

返回一个长度为 m 的数组 answer ，其中 answer[i] 是 nums 中 元素之和小于等于 queries[i] 的 子序列 的 最大 长度 。

子序列 是由一个数组删除某些元素（也可以不删除）但不改变剩余元素顺序得到的一个数组。

1.2.题目地址

https://leetcode.cn/problems/longest-subsequence-with-limited-sum/description/

2.解题方法

2.1.解题思路

二分查找

最长的子序列长度一定是最小的k个元素组成的序列，所以可以先升序排列nums，然后贪心的取前面的元素和

3.解题代码

python代码

class Solution:
    def answerQueries(self, nums: List[int], queries: List[int]) -> List[int]:
        # 思路：二分查找。最长的子序列长度一定是最小的k个元素组成的序列，所以可以先升序排列nums，然后贪心的取前面的元素和
        n = len(nums)
        nums.sort()
        preSumArr = [0] * (n + 1)
        for i in range(n):
            preSumArr[i + 1] = preSumArr[i] + nums[i]
        result = [0] * len(queries)
        for j, query in enumerate(queries):
            # preSumArr中查询最后一个小于或等于query的元素。红色区域：小于等于query的区域；红色区域：大于query的区域。l始终指向红色区域，r始终指向蓝色区域。最终的l即为result[j]
            l, r = -1, n + 1
            while l + 1 < r:
                mid = (r - l) // 2 + l
                if preSumArr[mid] <= query:
                    l = mid
                else:
                    r = mid
            result[j] = l
        return result

4.执行结果