LeetCode 2389. 和有限的最长子序列

最新推荐文章于 2024-06-10 17:05:04 发布

Sasakihaise_

最新推荐文章于 2024-06-10 17:05:04 发布

阅读量61

点赞数

分类专栏： LeetCode 文章标签：贪心二分

本文链接：https://blog.csdn.net/Sasakihaise_/article/details/129637227

版权

LeetCode 专栏收录该内容

438 篇文章 1 订阅

订阅专栏

2389. 和有限的最长子序列

【排序 + 前缀和 + 二分】首先子序列和顺序无关，要想让元素个数多就应该尽量先找小的元素。因此可以先把元素排序，然后计算前缀和与queries比较，找到第一个小于等于queries[i]的元素即可。

class Solution {

    public int binary(int[] arr, int k) {
        int l = 0, r = arr.length - 1, n = arr.length;
        while (l <= r) {
            int m = (l + r) >> 1;
            if (arr[m] > k) r = m - 1;
            else l = m + 1;
        }
        return r;
    }

    public int[] answerQueries(int[] nums, int[] queries) {
        int n = nums.length, sum = 0;
        Arrays.sort(nums);
        for (int i = 0; i < n; i++) {
            sum += nums[i];
            nums[i] = sum;
        }
        for (int i = 0; i < queries.length; i++) {
            queries[i] = binary(nums, queries[i]) + 1;
        }
        return queries;
    }
}