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<int>& price, vector<vector<int>>& special, vector<int>& 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]
*/