1、遍历
考虑到在本题中我们只需要或缺当前数组的子序列,并没有要求子序列必须是连续的。因此我们可以先对当前数组进行原地排序,而后我们从前向后利用二重循环添加需要的元素并判断当前元素的总和是否小于我们需要的queries[i]
即可。
class Solution {
public:
vector<int> answerQueries(vector<int> &nums, vector<int> &queries) {
vector<int> res;
sort(nums.begin(), nums.end());
for (int i: queries) {
int temp = 0;
int length = 0;
for (; length < nums.size() && temp <= i; ++length) {
temp += nums[length];
}
res.emplace_back((length == nums.size() && temp <= i) ? length : length - 1);
}
return res;
}
};
2、前缀和
我们可以对方法一进行优化,考虑到其中我们需要求连续数字和,我们可以将排序后的数组转化为前缀和数组,并利用二分查找进行搜索。
class Solution {
public:
vector<int> answerQueries(vector<int> &nums, vector<int> &queries) {
vector<int> res;
sort(nums.begin(), nums.end());
nums.insert(nums.begin(), 0);
for (int i = 1; i < nums.size(); ++i) {
nums[i] += nums[i - 1];
}
for (int i: queries) {
res.emplace_back(upper_bound(nums.begin(), nums.end(), i) - nums.begin() - 1);
}
return res;
}
};