No.66-LeetCode638-6维dp-选最优套装

起始dp状态是可以7进制压缩的,但是写起来容易错,倒不如这样写好了。

class Solution {
public:
    int shoppingOffers(vector<int>& price, vector<vector<int>>& special, vector<int>& needs) {
        const int base = 7;
        int dp[base][base][base][base][base][base];
        memset(dp,0,sizeof(dp));
        int N = price.size();
        int pc[7];
        memset(pc,0,sizeof(pc));
        for(int i=0;i<N;i++) pc[i] = price[i];
        int nd[7];
        memset(nd,0,sizeof(nd));
        for(int i=0;i<N;i++) nd[i] = needs[i];
        
        for(int a=0; a<=nd[0]; a++){
            for(int b=0; b<=nd[1]; b++){
                for(int c=0; c<=nd[2]; c++){
                    for(int d=0; d<=nd[3]; d++){
                        for(int e=0; e<=nd[4]; e++){
                            for(int f=0; f<=nd[5]; f++){
                                dp[a][b][c][d][e][f] = a*pc[0]+b*pc[1]+c*pc[2]+d*pc[3]+e*pc[4]+f*pc[5];
                                for(int i=0;i<special.size();i++){
                                    int sp[7];
                                    memset(sp,0,sizeof(sp));
                                    for(int j=0;j<N;j++) sp[j] = special[i][j];
                                    int key = special[i][N];
                                    if(a>=sp[0]&&b>=sp[1]&&c>=sp[2]&&d>=sp[3]&&e>=sp[4]&&f>=sp[5]){
                                        dp[a][b][c][d][e][f] = min(dp[a][b][c][d][e][f],dp[a-sp[0]][b-sp[1]][c-sp[2]][d-sp[3]][e-sp[4]][f-sp[5]] + key);
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
        return dp[nd[0]][nd[1]][nd[2]][nd[3]][nd[4]][nd[5]];
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值