Leetcode 638. 大礼包 C++

Leetcode 638. 大礼包

题目

在LeetCode商店中， 有许多在售的物品。

然而，也有一些大礼包，每个大礼包以优惠的价格捆绑销售一组物品。

现给定每个物品的价格，每个大礼包包含物品的清单，以及待购物品清单。请输出确切完成待购清单的最低花费。

每个大礼包的由一个数组中的一组数据描述，最后一个数字代表大礼包的价格，其他数字分别表示内含的其他种类物品的数量。

任意大礼包可无限次购买。

测试样例

示例 1:

输入: [2,5], [[3,0,5],[1,2,10]], [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:

输入: [2,3,4], [[1,1,0,4],[2,2,1,9]], [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更加便宜。

说明:

1. 最多6种物品， 100种大礼包。
2. 每种物品，你最多只需要购买6个。
3. 你不可以购买超出待购清单的物品，即使更便宜。

题解

深搜dfs回溯
对于第i种礼包我们可以选择要和不要，显然就是深搜回溯算法。再进行选择前，我们先假设不用礼包，只买单个，这样我们可以得到一个结果。之后便是对于大礼包的选择，使用dfs回溯。详细过程见代码

代码

	vector<int> onePrice;	//单买的各个物品的加个	
    vector<vector<int>> manyPrice;		//礼包种各个物品的数量及礼包价格
    int ans = INT_MAX;
    void dfs(int now,vector<int>& needs){		//now为当前的花费
        if(now >= ans)  return;			//剪枝
        int all = now;
        for(int i=0; i<needs.size(); i++){			//只考虑单买
            if(needs[i]<0)  return;			//有的物品超额购买，直接return，不考虑此方案
            if(needs[i]>0)  all += needs[i]*onePrice[i];  
        }
        ans = min(ans,all);
        if(all == now)  return;
        for(int i=0; i<manyPrice.size(); i++){		//考虑第i种大礼包的购买
            int j;
            for(j=0; j<needs.size(); j++)
                needs[j] -= manyPrice[i][j];
            now += manyPrice[i][j];
            dfs(now,needs);
            for(j=0; j<needs.size(); j++)
                needs[j] += manyPrice[i][j];
            now -= manyPrice[i][j];
        }
    }
    int shoppingOffers(vector<int>& price, vector<vector<int>>& special, vector<int>& needs) {
        onePrice = price;
        manyPrice = special;
        dfs(0,needs);
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shopping-offers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。