题目描述
小力将 N 个零件的报价存于数组 nums。
小力预算为 target,假定小力仅购买两个零件,要求购买零件的花费不超过预算,请问他有多少种采购方案。
注意:答案需要以 1e9 + 7 (1000000007) 为底取模,如:计算初始结果为:1000000008,请返回 1
示例 1:
输入:nums = [2,5,3,5], target = 6
输出:1
解释:预算内仅能购买 nums[0] 与 nums[2]。
示例 2:
输入:nums = [2,2,1,9], target = 10
输出:4
解释:符合预算的采购方案如下:
nums[0] + nums[1] = 4
nums[0] + nums[2] = 3
nums[1] + nums[2] = 3
nums[2] + nums[3] = 10
提示:
2 <= nums.length <= 10^5
1 <= nums[i], target <= 10^5
算法1
(双指针) $O(n)$
Leetcode A+B的大数据版本
将数组排序,l r两指针指向头尾
1 由于数组有序 ,如果l r的元素和小于target,
那么l 和 l+1~r之间的零件搭配都会小于预算target,计入答案。
然后l++,计算下一个零件确定后,第二个零件有多少个可选择的数目
2 如果l和r的元素和大于target,那么说明零件价格过大,需要减少零件的价格,r–
直到达到情况1的条件
C++ 代码
class Solution {
public:
long long ans = 0;
int purchasePlans(vector<int>& nums, int target) {
sort(nums.begin(),nums.end());
int l= 0; int r = nums.size()-1;
while(l<r){
if(nums[l]+nums[r] > target){
r--;
}else if(nums[l]+nums[r] <= target){
cout << l<<" " << r<<endl;
ans += r-l;
ans = ans% 1000000007;
l++;
}
}
return ans;
}
};