OJ刷题--大礼包(C++含输入输出完整代码)

题目

在 LeetCode 商店中, 有 n 件在售的物品。每件物品都有对应的价格。然而,也有一些大礼包,每个大礼包以优惠的价格捆绑销售一组物品。
给你一个整数数组 price 表示物品价格,其中 price[i] 是第 i 件物品的价格。另有一个整数数组 needs 表示购物清单,其中 needs[i] 是需要购买第 i 件物品的数量。
还有一个数组 special 表示大礼包,special[i] 的长度为 n + 1 ,其中 special[i][j] 表示第 i 个大礼包中内含第 j 件物品的数量,且 special[i][n] (也就是数组中的最后一个整数)为第 i 个大礼包的价格。
返回 确切 满足购物清单所需花费的最低价格,你可以充分利用大礼包的优惠活动。你不能购买超出购物清单指定数量的物品,即使那样会降低整体价格。任意大礼包可无限次购买。
示例 1:
输入:price = [2,5], special = [[3,0,5],[1,2,10]], needs = [3,2]
输出:14
解释:有 A 和 B 两种物品,价格分别为 ¥2 和 ¥5 。
大礼包 1 ,你可以以 ¥5 的价格购买 3A 和 0B 。
大礼包 2 ,你可以以 ¥10 的价格购买 1A 和 2B 。
需要购买 3 个 A 和 2 个 B , 所以付 ¥10 购买 1A 和 2B(大礼包 2),以及 ¥4 购买 2A 。
示例 2:
输入:price = [2,3,4], special = [[1,1,0,4],[2,2,1,9]], needs = [1,2,1]
输出:11
解释:A ,B ,C 的价格分别为 ¥2 ,¥3 ,¥4 。
可以用 ¥4 购买 1A 和 1B ,也可以用 ¥9 购买 2A ,2B 和 1C 。
需要买 1A ,2B 和 1C ,所以付 ¥4 买 1A 和 1B(大礼包 1),以及 ¥3 购买 1B , ¥4 购买 1C 。
不可以购买超出待购清单的物品,尽管购买大礼包 2 更加便宜。
提示:
n == price.length
n == needs.length
1 <= n <= 6
0 <= price[i] <= 10
0 <= needs[i] <= 10
1 <= special.length <= 100
special[i].length == n + 1
0 <= special[i][j] <= 50

思路

参考链接 大礼包

代码(C++完整代码含输入输出)

#include
#include
#include
#include
using namespace std;
class Solution {
public:
    int shoppingOffers(vector<int>& price, vector<vector<int>>& special, vector<int>& needs)
    {
        int n = needs.size();
        if (0 == n)
        {
            return 0;
        }
        int m = special.size();
        int ans = 0;
        for (int i = 0; i < n; i++)
        {
            ans += needs[i] * price[i];  //如果用不了下面的大礼包,那么需要单独购买
        }                            //如果能用大礼包,取两者较小值
        for (int i = 0; i < m; i++)
        {
            int flag = 1;
            for (int j = 0; j < n; j++)
            {
                if (special[i][j] > needs[j])
                {
                    flag = 0;
                    break;
                }
            }
            if (flag)
            {
                for (int j = 0; j < n; j++)
                {
                    needs[j] -= special[i][j];
                }
                int val = special[i][n] + shoppingOffers(price, special, needs);
                ans = min(val, ans);
                for (int j = 0; j < n; j++)
                {
                    needs[j] += special[i][j];
                }
            }
        }
        return ans;
    }
};
int main()
{
    Solution sol;
    vector<int> price;
    vector<vector<int>> special;
    vector<int> needs;
    int i;
    while (cin >> i)
    {
        price.push_back(i);
        if (cin.get() == '\n') break;
    }
    int temp;
    vector<int> v;
    special.clear();
    for (int k = 0; k < 2; k++)
    {
        v.clear();
        int t;
        while (cin >> t)
        {
            v.push_back(t);
            if (cin.get() == '\n') break;
        }
        special.push_back(v);
    }

    int j;
    while (cin >> j)
    {
        needs.push_back(j);
        if (cin.get() == '\n') break;
    }
    cout << sol.shoppingOffers(price, special, needs) << endl;
    return 0;
}

你可能感兴趣的:(OJ刷题,c++,算法,leetcode)