题目描述
小力将 N 个零件的报价存于数组 nums。小力预算为 target，假定小力仅购买两个零件，要求购买零件的花费不超过预算，请问他有多少种采购方案。 注意：答案需要以 1e9 + 7 (1000000007) 为底取模，如：计算初始结果为：1000000008，请返回 1
Example 1:
输入：nums = [2,5,3,5], target = 6 输出：1 解释：预算内仅能购买 nums[0] 与 nums[2]。
Example 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
Constraints:
- 2 <= nums.length <= 10^5
- 1 <= nums[i], target <= 10^5
解题思路
- 用暴搜会超时
- 此时可以先排序，如果从后往前，找到一个位置，num[pos] + num[i] > target的时候，证明仍然不能作为一个采购方案，需要再次往前查找。
C++代码
class Solution {
public:
int purchasePlans(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
long long res = 0;
long long mod = 1000000007;
int j = nums.size() - 1;
for(int i = 0; i < j; i ++){
while(i < j && nums[i] + nums[j] > target){
j--;
}
res += j - i;
}
return res % mod;
}
};
