2389. 和有限的最长子序列

题目描述

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

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

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

示例 1：

输入：nums = [4,5,2,1], queries = [3,10,21]
输出：[2,3,4]
解释：queries 对应的 answer 如下：

• 子序列 [2,1] 的和小于或等于 3 。可以证明满足题目要求的子序列的最大长度是 2 ，所以 answer[0] = 2 。
• 子序列 [4,5,1] 的和小于或等于 10 。可以证明满足题目要求的子序列的最大长度是 3 ，所以 answer[1] = 3 。
• 子序列 [4,5,2,1] 的和小于或等于 21 。可以证明满足题目要求的子序列的最大长度是 4 ，所以 answer[2] = 4 。
  示例 2：

输入：nums = [2,3,4,5], queries = [1]
输出：[0]
解释：空子序列是唯一一个满足元素和小于或等于 1 的子序列，所以 answer[0] = 0

分析

先将nums数组按升序排列，然后遍历queries数组，对每一个取出的元素都将遍历nums数组，也就是2层for循环，比较该元素与nums数组的前n项和即可。

代码

class Solution {
    public int[] answerQueries(int[] nums, int[] queries) {
    	//排序
        List<Integer> list = Arrays.stream(nums).boxed().collect(Collectors.toList());
		Collections.sort(list);
		int[] result = new int[queries.length];
		int s=0;//s用来作为nums数组前i项和
		//比较
		for(int j=0;j<queries.length;j++){
			for(int i=0;i<list.size();i++){
				s+=list.get(i);
				if(s>queries[j]){
					result[j]=i;
					s=0;
					break;
				}else{
					//考虑到nums数组所有元素和都小于queries[j]的情况
					if(i==list.size()-1){
						result[j]=list.size();
                        s=0;
						break;
					}
				}
			}
		}
		return result;
    }
}