487. 金明的预算方案 分组背包 + 二进制枚举 动态规划

题目

题解思路

对于每一组的物品 ，如果我们要选择就必须先选主物品，再选择其他的。
因为附属品的个数不超过2个

所以我们可以进行二进制枚举 2的2次方的复杂度 。

这些来看就是分组背包的问题了。

我们枚举的其实就是决策。在每个主物品下的每个体积的情况下枚举决策出最优情况

这里vector二维pair的运用也挺好的

参考文章和闫总视频

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;

typedef pair <int , int > PII ;
#define  x  first
#define  y  second
const  int  INF =  0x3f3f3f3f;

int f[40100] ;
PII master[110];
vector <PII> ser[110];

int main ()
{
    ios::sync_with_stdio(false);
    int m,n;
    cin>>m>>n;
    for (int i = 1 ; i <= n ; i++ )
    {
        int v,p,q;
        cin>>v>>p>>q;
        if (!q)
            master[i] = {v , v*p };
        else
            ser[q].push_back({ v , v*p });
    }
    for (int i = 1 ; i <= n ; i++ )
    {
        for (int j = m ; j >= 0 ; j-- )
        {
            for (int k = 0 ; k < 1 << ser[i].size() ; k++ )
            {
                int a = master[i]. x , b = master[i].y ;
                for (int u = 0 ; u < ser[i].size() ; u++ )
                {
                    if ( k >> u & 1 )
                    {
                        a += ser[i][u].x ;
                        b += ser[i][u].y ;
                    }
                }
                if ( j >= a )
                    f[j] = max( f[j] , f[j-a] + b );
            }
        }
    }
    cout<<f[m]<<"\n";
    return 0 ;
}