力扣(LeetCode)2389. 和有限的最长子序列(C++)

贪心

想要子序列的长度最长，就要子序列的和最小，显然从小到大选取数组数字最合适。本题的子序列是可间断的数组元素（子序列不一定连续），数组元素的顺序不影响子序列的和。对数组 nums 排序，就很方便从小到大选数了。

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());
        int sum = 0;
        vector<int> ans(m);
        for (int i = 0, j = 0; i < m; i ++) {
            while (j < n && sum <= queries[i]) sum += nums[j ++];
            while (sum > queries[i]) sum -= nums[-- j];
            ans[i] = j;
        }
        return ans;
    }
};

1. 时间复杂度 : $O(nlogn+n\times m)$ ， $n$ 是数组 $nums$ 的大小，$m$ 是数组 $queries$ 的大小，排序 $nums$ 的时间复杂度是 $O(nlogn)$ ，枚举查询和数字的时间复杂度 $O(n\times m$)，总体时间复杂度$O(nlogn+n\times m)$ 。
2. 空间复杂度 : $O(logn)$ ， 不考虑答案使用的空间，快排的空间复杂度 $O(logn)$ 。

前缀和 + 二分查找

枚举 $nums$ 数组，其实是按顺序增减数字，也就是看某个数组下标的前缀和，和查询做比较。直接构造前缀和数组，二分查找前缀和数组，以加速枚举数组 $nums$ 的效率。

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());
        int sum = 0;
        vector<int> p(n + 1);
        for (int i = 1; i <= n; i ++) p[i] = p[i - 1] + nums[i - 1];
        vector<int> ans(m);
        for (int i = 0, j = 0; i < m; i ++) {
            int l = 0, r = n - 1;
            while(l <= r) {
                int mid = l + (r - l >> 1);
                if (p[mid + 1] > queries[i]) r = mid - 1;
                else l = mid + 1;
            }
            ans[i] = l;
        }
        return ans;
    }
};

1. 时间复杂度 : $O(n\times (logn+logm))$ ， $n$ 是数组 $nums$ 的大小，$m$ 是数组 $queries$ 的大小，排序 $nums$ 的时间复杂度是 $O(nlogn)$ ，枚举查询和二分查找的时间复杂度 $O(nlogm$)，总体时间复杂度$O(n\times (logn+logm))$ 。
2. 空间复杂度 : $O(n)$ ， 不考虑答案使用的空间，前缀和数组的空间复杂度 $O(n)$ 。

AC

时空对比。

致语

• 理解思路很重要
• 读者有问题请留言，清墨看到就会回复的。