AcWing 487. 金明的预算方案

#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#define v first
#define w second
using namespace std;
typedef pair<int, int> PII;
const int N = , M = ;
int n, m;
PII master[N];
vector<PII> servent[N];
int f[M];
int main() {
    cin >> m >> n;//m总钱数  n个数
    for (int i = ; i <= n; i ++ ) {
        int v, p, q;
        cin >> v >> p >> q;
        p *= v;//价值
        if (!q) master[i] = {v, p};//如果是0，表示为主件
        else servent[q].push_back({v, p});//如果不是0，为附件，插入相关主键
    }
    for (int i = ; i <= n; i ++ )//个数
        for (int u = m; u >= ; u -- ) {//总钱数
            for (int j = ; j <  << servent[i].size(); j ++ )
            //转换为二进制，对于每个主件，有几种选法
             {
                int v = master[i].v, w = master[i].w;
                for (int k = ; k < servent[i].size(); k ++ )
                    if (j >> k & )
                    {//如果第k位是1，那就说明选第k个
                        v += servent[i][k].v;
                        w += servent[i][k].w;
                    }
                if (u >= v) f[u] = max(f[u], f[u - v] + w);
            }
        }
    cout << f[m] << endl;
    return ;
}

#include<iostream>
#include<vector>
#define v first
#define w second
using namespace std ;
typedef pair<int,int>PII;
const int N=;
int f[N];
int n,m;
vector<PII>servent[N];
PII master[N];
int main()
{
    cin>>m>>n;
    for(int i=;i<=n;i++)
    {
        int v,p,q;
        cin>>v>>p>>q;
        p*=v;
        if(!q)
            master[i]={v,p};
        else
            servent[q].push_back({v,p});
    }
    for(int i=;i<=n;i++)
        for(int j=m;j>=;j--)
            for(int k=;k< <<servent[i].size();k++)
            {
                int w=master[i].w,v=master[i].v;
                for(int u=;u<servent[i].size();u++)
                {
                    if(k>>u&)
                    {
                        w+=servent[i][u].w;
                        v+=servent[i][u].v;
                    }
                }
                if(j>=v)
                    f[j]=max(f[j],f[j-v]+w);
            }
    cout<<f[m]<<endl;
    return ;
}