Java实现LeetCode 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; } }
版权声明:本文为a1439775520原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。