LeetCode 638. Shopping Offers(入门 状压dp)

给出n种商品的价格以及需求量，以及一些套餐的优惠价，问最少花多少钱，任何一种商品都不能多买。

不超过6种商品，每种商品需求不超过6个，但是可能有100种套餐。

leetcode感觉题目普遍偏简单一点，状压dp竟然定义难度是中等，不知道是不是我搞麻烦了。

每种商品不超过6个，那么需要占三位，6种商品就是18位（这样相当于使用8进制数，也可以自己计算6进制数，那样可以更节省空间），注意要尽可能的避免无效运算，还是有可能会超时的。而且要注意溢出问题，就是一种商品实际购买如果超过8了那就会溢出影响其他商品的数量。

class Solution {
public:
    int dp[1000001];
    int mx = 1000000;
    int shoppingOffers(vector& price, vector>& special, vector& needs) {
        int ans = 0;
        for(int i = 0; i < needs.size(); i++) {
            ans += needs[i] << (i * 3);
        }
        mx = ans;
        memset(dp, 0x3f, sizeof(dp));
        dp[0] = 0;
        for(int i = 0; i < mx; i++) {
            if(dp[i] == 0x3f3f3f3f) continue;
            for(int j = 0; j < price.size(); j++) {
                int x = i + (1 << (j * 3));
                if(((x >> (j * 3)) & 7) > needs[j]) continue;
                if(x > mx) continue;
                dp[x] = min(dp[x], dp[i] + price[j]);
            }
            for(int j = 0; j < special.size(); j++) {
                int x = i, k;
                for(k = 0; k < special[j].size() - 1; k++) {
                    if(((x >> (k * 3)) & 7) + (special[j][k]) > needs[k]) {
                        x = mx + 1;
                        break;
                    }
                    x += special[j][k] << (k * 3);
                }
                
                if(x > mx) continue;
                //printf("%2o %2o %3d\n", i, x, j);
                dp[x] = min(dp[x], dp[i] + special[j][k]);
            }
        }
        
        //printf("%o\n", ans);
        //for(int i = 0; i <= ans; i++) {
        //    printf("%2o %d\n", i, dp[i]);
        //}
        return dp[ans];
    }
};
/*
[4,10,1,5,5,3]
[[1,2,3,3,4,1,8],[3,4,5,5,5,2,14],[2,4,5,1,1,3,22]]
[1,6,5,1,1,4]
*/