小力将 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

Python 解答:
1.暴力超时

class Solution:
    def purchasePlans(self, nums: List[int], target: int) -> int:
        total = 0
        for i in range(len(nums)):
            for j in range(i+1, len(nums)):
                if nums[i]+nums[j] <= target:
                    total += 1
        return total

2.双指针

class Solution:
    def purchasePlans(self, nums: List[int], target: int) -> int:
        nums.sort()
        total = 0
        i, j = 0, len(nums)-1
        while i < j:
            if nums[i]+nums[j] > target:
                j -= 1
            else:
                total += j-i
                i += 1
        return total % (1000000007)
最后修改日期: 2021年6月12日

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。