Every day a Leetcode
题目来源:2389. 和有限的最长子序列
解法1:排序+贪心
由题意可知, nums 的元素次序对结果无影响,因此我们对 nums 从小到大进行排序。
基于贪心的思想,和有限的最长子序列由最小的前几个数组成。
遍历 queries,假设当前查询值为 q。
对于每个查询值 q,初始化和 sum=0,从头开始遍历排序后的 nums 数组,sum 累加 nums[i] 直至 sum>q 或者遍历完数组 nums,最后更新 answer 数组。
遍历完 queries,answer 数组即为答案。
代码:
/*
* @lc app=leetcode.cn id=2389 lang=cpp
*
* [2389] 和有限的最长子序列
*/
// @lc code=start
class Solution
{
public:
vector<int> answerQueries(vector<int> &nums, vector<int> &queries)
{
int n = nums.size();
int m = queries.size();
sort(nums.begin(), nums.end());
vector<int> answer(m);
for (int i = 0; i < m; i++)
{
int sum = 0;
int count = 0;
int index = 0;
while (index < n && sum <= queries[i])
{
sum += nums[index];
index++;
if (sum <= queries[i])
count++;
}
answer[i] = count;
}
return answer;
}
};
// @lc code=end
结果:
复杂度分析:
时间复杂度:O(n*(logn+m)),其中 n 是数组 nums 的长度,m 是数组 queries 的长度。对 nums 进行排序需要 O(nlogn) 的时间。
空间复杂度:O(1),我们只使用了常数个变量。返回值不计入空间复杂度。