分组背包问题,转化为0-1背包问题

购物单

http://www.nowcoder.com/questionTerminal/f9c6f980eeec43ef85be20755ddbeaf4

看了各个大佬的分析,终于写出了我自己的代码,其实知道了如何进行状态转移,就基本大差不差

#include <bits/stdc++.h>

using namespace std;

int main()
{
    // 带条件的0-1背包问题
    // 创建dp数组
    int N(0), m(0);
    cin >> N >> m;
    // N是总预算,相当于背包问题的容量
    int tv(0), tp(0), tq(0);
    //vector<vector<int>> dp(m+1, vector<int>(N/10+1, 0));
    vector<vector<int>> primary;
    map<int, vector<vector<int>>> sub;
    map<int, int> linenumtomainnum;
    int i=0;
    while (i<m) {
        cin >> tv >> tp >> tq;
        if (tq == 0)
        {
            primary.push_back(vector<int>{tv, tp, tq});
            linenumtomainnum[i+1] = primary.size() - 1;
        }
        else {
            sub[linenumtomainnum[tq]].push_back(vector<int>{tv, tp, tq});
        }
        i++;
    }
    // 有mainnum个主件
    // 初始化
    vector<vector<int>> w(primary.size(), vector<int>{});
    vector<vector<int>> v(primary.size(), vector<int>{});
    for (int i = 0; i<primary.size(); i++) {
        int subsize = sub[i].size();    // 附件的数量
        int wp = primary[i][0];
        int vp = primary[i][0] * primary[i][1];        // 主件的w(价格)和v(价格乘重要程度)
                                                       // 加入只有主件的情况
        w[i].push_back(wp);
        v[i].push_back(vp);
        if (subsize == 0)    // 如果没有继续向下,说明存在至少一个附件
            continue;
        if (subsize == 1)    // 只有一个附件
        {
            int ws1 = sub[i][0][0];    // 附件的价格
            int vs1 = sub[i][0][0] * sub[i][0][1];    // 附件的价格乘重要程度
                                                      // 主件+附件的模式
            w[i].push_back(ws1 + wp);
            v[i].push_back(vs1 + vp);
        }
        if (subsize == 2)    // 有两个附件,附件1+附件2
        {
            int ws1 = sub[i][0][0];    // 附件的价格
            int vs1 = sub[i][0][0] * sub[i][0][1];    // 附件的价格乘重要程度
            int ws2 = sub[i][1][0];    // 附件的价格
            int vs2 = sub[i][1][0] * sub[i][1][1];    // 附件的价格乘重要程度
                                                      // 主件+附件1
            w[i].push_back(wp + ws1);
            v[i].push_back(vp + vs1);
            // 主件+附件2
            w[i].push_back(wp + ws2);
            v[i].push_back(vp + vs2);
            // 主件+附件1+附件2
            w[i].push_back(wp + ws1 + ws2);
            v[i].push_back(vp + vs1 + vs2);
        }
    }
    // 动态规划开始
    vector<vector<int>> dp(w.size() + 1, vector<int>(N / 10 + 2, 0));
    for (int i = 1; i <= w.size(); i++) {
        int ti = i - 1;
        for (int j = 0; j <= N+10; j += 10) {
            int tj = j / 10;
            dp[i][tj] = dp[i - 1][tj];
            for (int k = 0; k<w[ti].size(); k++) {
                if(j>=w[ti][k])
                    dp[i][tj] = max(dp[i][tj], dp[i - 1][(j-w[ti][k])/10] + v[ti][k]);
            }
        }
    }
    cout << dp[w.size()][N/10+1] << endl;
    return 0;
}

不过我的代码没有经过优化,应该能用一维数组来做的,不过先不管了,菜鸡如我还是先学习背包问题的套路吧,优化后面再说

全部评论

相关推荐

04-16 10:27
已编辑
美团_Saas_后端开发
今天周一休息,突发奇想写一篇阶段总结。如题,我已经去了一个和Java彻底毫无关联的行业。曾经我以为自己能在计算机行业发光发热,拿到美团offer那会感觉自己天都亮了。没想到刚入行一年多就当了逃兵。从最开始的热爱到现在一看到代码就厌恶,不知道自己经历了什么。所以我去干什么了?答案是:在成都当了租房销售。上班那会压力大了就念叨着去干租房中介,但是一直下不去这个决心,想着自己学了四年多的计算机知识,终究还是不甘心。终于在某一天准备八股文的时候,看着无数篇和工作内容关系不大的理论知识,那一刻下定决心,决定尝试一下销售行业,也算是给自己一个交代。后面阴差阳错的投了成都自如去当租房管家,没想到面试很顺利,在当天一百多个面试的人里面,我成为了为数不多通过的几个幸运儿之一。目前已经培训通过,正式入职,也开了单,有压力但是每天过得很开心,真心喜欢那种和人交流的感觉,哪怕是最后没有选择找我租房。说这些也是想告诉那些大三,大四正在找Java实习而焦虑的同学:你们现在还年轻,选择很多,容错率也很高,可以尽情去尝试自己喜欢的行业和工作。不用因为某一次的面试没通过或者简历石沉大海而焦虑,更不用因为身边人都在挤编程的独木桥就强迫自己跟风。也算是自己的碎碎念吧,也希望自己能在新的领域取得一点小成就。也祝牛油工作顺利!
沉淀小子:干啥都不丢人啊,生存是必须要的,销售很考验一个人综合素质能力的,好的销售人脉和资源可不比写字楼的白领差啊
点赞 评论 收藏
分享
评论
4
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务