Java实现LeetCode 5450. 满足条件的子序列数目(双指针)

5450. 满足条件的子序列数目

 

给你一个整数数组 nums 和一个整数 target 。

请你统计并返回 nums 中能满足其最小元素与最大元素的 和 小于或等于 target 的 非空 子序列的数目。

由于答案可能很大，请将结果对 10^9 + 7 取余后返回。

 

示例 1：

输入：nums = [3,5,6,7], target = 9
输出：4
解释：有 4 个子序列满足该条件。
[3] -> 最小元素 + 最大元素 <= target (3 + 3 <= 9)
[3,5] -> (3 + 5 <= 9)
[3,5,6] -> (3 + 6 <= 9)
[3,6] -> (3 + 6 <= 9)

示例 2：

输入：nums = [3,3,6,8], target = 10
输出：6
解释：有 6 个子序列满足该条件。（nums 中可以有重复数字）
[3] , [3] , [3,3], [3,6] , [3,6] , [3,3,6]

示例 3：

输入：nums = [2,3,3,4,6,7], target = 12
输出：61
解释：共有 63 个非空子序列，其中 2 个不满足条件（[6,7], [7]）
有效序列总数为（63 - 2 = 61）

示例 4：

输入：nums = [5,2,4,1,7,6,8], target = 16
输出：127
解释：所有非空子序列都满足条件 (2^7 - 1) = 127

 

提示：

• 1 <= nums.length <= 10^5
• 1 <= nums[i] <= 10^6
• 1 <= target <= 10^6

这里也有一个好理解得方法

import java.math.BigInteger;
class Solution {
     public int numSubseq(int[] nums, int target) {
        int n=nums.length;
        Arrays.sort(nums);
        int index=n-1;
        BigInteger two=BigInteger.valueOf(2),m=BigInteger.valueOf(1000000007),sum=BigInteger.ZERO;
        for(int i=0;i<n;i++){
            while(index>=0 && nums[i]+nums[index]>target) index--;
            if(index<i) break;
            sum=sum.add(two.pow(index-i)).mod(m);
        }
        return sum.intValue();
    }
}

 

 

这个是效率高得代码

 

import java.math.BigInteger;
class Solution {
     public int numSubseq(int[] nums, int target) { 
         //先给数组排个序
         Arrays.sort(nums);
         //看是不是最小值是小于target的，如果最小值都比他小，
         //那么直接返回就好了，不存在结果
        if (nums[0] * 2 > target) {
            return 0;
        }
        //双指针写
        int left = 0;
        int right = nums.length - 1;
        //声明一个变量取记录答案
        int res = 0;
        //pow数组是记录2的下标次方
        int[] pow = new int[nums.length];
        pow[0] = 1;
        //结果取余
        int mode = 1_0000_0000_7;
        //把二进制的数放进数组
        for (int i = 1; i < nums.length; i ++) {
            pow[i] = pow[i-1] * 2;
            pow[i] %= mode;
        }
        //双指针
        while (left <= right) {
            //左面的和右面得和小于target才可以
            if (nums[left] + nums[right] <= target) {
                //这里的意思是，当我有3个值得时候，3个值得非空子序列个数为3得二次方-1
                //有几个值，就是right-left个值，
                res += pow[right - left];
                //取余
                res %= mode;
                //左指针向右走一位
                left ++;
            }
            //如果不符合条件得话，右指针向左走1位
            else {
                right --;
            }
        }
        return res;
    }
 
}