采购方案---java
小力将 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
提示:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4xy4Wx
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
nums:[1,2,2,9] target: 10
我一开始写了两个for循环,这显然时间复杂度是O(n^2),消耗的时间太多,通不过。
1.如果我们首先进行对数组排序Arrays.sort(nums)
2.我们使用i,j双指针,i为头,j为尾,res来记录有多少种方案
3.当i=0,j=3,nums[i]+nums[j]<=target,我们可以j-i种方案,存储到res中
4.此时我们要i++,i=1,j=2,nums[i]+nums[j]<=target,则又有j-i种方案,存储到res
5.最后输出res,这个时间复杂度只有一个while,所以复杂度尾O(n)
class Solution {
public int purchasePlans(int[] nums, int target) {
Arrays.sort(nums);
int i =0;
int res=0;
int j =nums.length-1;
while(i<j){
if(nums[i]+nums[j]<=target){
res=res%1000000007;
res += j-i;
i++;
}else{
j--;
}
}
return res;
}
}