Problem: 2389. 和有限的最长子序列
思路
初看此问题感觉毫无思路,但看到题目没有对nums数组下标有要求,子序列也对顺序没有要求,那么首先想到就是给nums排序再说。有序后
“nums 中 元素之和小于等于 queries[i] 的 子序列 的 最大长度“ 就 可以等效为:
求nums前j项刚好小于等于queries[i] 的j值。
这样就可以将nums排序后,遍历queries 元素与nums中前j项和进行比较,找到前j项和满足小于等于queries[i],然后添加到res列表中传回。
然后再考虑列表长度为1和为0的边界条件。
Code
class Solution:
def answerQueries(self, nums: List[int], queries: List[int]) -> List[int]:
nums.sort()
n = len(nums)
m = len(queries)
res = []
if m == n ==1 and nums[0] == queries[0]:# 第一次提交后有一个用例nums=[100000] ,queries=[100000],输出是[1]
return [1]
if m <= 1:
return [0]
for i in range(m):
sum = 0
j = 0
while j < n and sum <= queries[i]:
sum += nums[j]
j +=1
if j < n or sum > queries[i]: #此处j值返回是否减1需要注意调试,易出错的地方
j-=1
res.append(j)
return res
后记
查看其他题解后发现基本都用到了二分查找,表示不理解。努力理解为嘛要用二分查找。。。