leetcodeLCP 28. 采购方案

最新推荐文章于 2024-10-01 18:52:56 发布

原创

最新推荐文章于 2024-10-01 18:52:56 发布 · 187 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#动态规划 #双指针

这道题目要求在不超过预算target的情况下，从N个零件报价nums中找出所有两件商品组合的采购方案数量。通过动态规划和双指针两种方法解决，注意结果需对1e9 + 7取模。

题目链接：https://leetcode-cn.com/problems/4xy4Wx/

题意：

小力将 N 个零件的报价存于数组 nums。小力预算为 target，假定小力仅购买两个零件，要求购买零件的花费不超过预算，请问他有多少种采购方案。

注意：答案需要以 1e9 + 7 (1000000007) 为底取模，如：计算初始结果为：1000000008，请返回 1

方法一：动态规划，枚举以i开头有几种方案

class Solution {
public:
    int purchasePlans(vector<int>& nums, int target) {
        const int MOD = 1e9+7;//余数
        sort(nums.begin(),nums.end());//排序一下
        int size = nums.size();//统计向量的长度
        vector<int> dp(size+1,0);//动态规划数组
        for(int i=0;i<size;i++)//遍历枚举每一个元素
        {
            int index = upper_bound(nums.begin(),nums.end(),target-nums[i])-nums.begin();//找到第一个大于target-nums[i]的数
            if(index<=i) break;//如果该数小于等于i就说明重复了，没有必要再找了
            dp[i] = index-i-1;//返回这样的组合方案有多少
        }
        int ans = 0;//最后的答案数目
        for(int