1498. 满足条件的子序列数目-快速排序+二分查找+快速幂-力扣双百代码
给你一个整数数组 nums 和一个整数 target 。
请你统计并返回 nums 中能满足其最小元素与最大元素的 和 小于或等于 target 的 非空 子序列的数目。
由于答案可能很大，请将结果对 109 + 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）
这一题可能比较复杂，但其实分解之后还是很简单的，先排序，排序之后，我们定标-就是针对每一个集合一定会出现某个元素，然向里面添加可能能添加的元素，解题代码如下：
void quick(int *a,int low,int high){
if(low<high){
int l=low,h=high,p=a[low];
while(low<high){
while(low<high&&a[high]>=p){
high--;
}
a[low]=a[high];
while(low<high&&a[low]<=p){
low++;
}
a[high]=a[low];
}
a[low]=p;
quick(a,l,low-1);
quick(a,low+1,h);
}
}
int find_b_search(int *nums,int low,int high,int target){
while(low<=high){
int mid=(low+high)/2;
if(nums[mid]<=target){
low=mid+1;
}
else{
high=mid-1;
}
}
return low-1;
}
int my_pow(int n){
int re=1;
for(int i=0;i<n;i++){
re=re*2%1000000007;
}
return re;
}
int numSubseq(int* nums, int numsSize, int target){
quick(nums,0,numsSize-1);
int dp[numsSize];
dp[0]=1;
for(int i=1;i<numsSize;i++){
dp[i]=dp[i-1]*2%1000000007;
}
int count=0;
for(int i=0;i<numsSize&&nums[i]*2 <= target;i++){
int index=find_b_search(nums,0,numsSize-1,target-nums[i]);
if(index>=i){
count=(count+dp[index-i])%1000000007;
}
}
return count;
}
|