题目要求:
给你一个长度为 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 。
思路:
对数组nums进行排序,从小到大排好。如示例1,nums排好后是1,2,4,5,从nums的第一位开始相加,直到超过数组queries中的第 i 位,相加时要统计相加的次数,该次数就是数组answer的第 i 位数值。如queries的第一位是3,从0开始加nums的每一位,1<3,count+1=1;1+2<3, count+1=2; 1+2+4>3, 结束本次循环,满足要求的子序列最大长度就是2,answer[0]就是2。
代码:
public class AnswerQueries {
public static void main(String[] args) {
int[] nums = {2,3,4,5};
int[] queries={1,2};
int[] answer=answerQueries(nums,queries);
for(int i=0;i<answer.length;i++){
System.out.print(answer[i] );
}
}
public static int[] answerQueries(int[] nums, int[] queries){
int[] answer = new int[queries.length];
for(int i = nums.length-1;i>0;i--){ //对nums排序
for(int j = i-1;j>=0;j--){
if(nums[j]>nums[i]){
int temp = nums[i];
nums[i]=nums[j];
nums[j]=temp;
}
}
}
for(int i=0;i<queries.length;i++){
int sum=0;
int count = 0;
for(int j=0;j<nums.length;j++){
if(sum<=queries[i]){
sum+=nums[j];
if(sum>queries[i]){
break;
}
count++;
}
}
answer[i] = count;
}
return answer;
}
}
注意:遍历queries数组时,对每一位数都要重新置sum 和 count 为0。