【分组背包】P1757 通天之分组背包

在记录每组有哪几种物品的时候，我是开了map,组号映射到一个vector，vector里面存着该组的物品的下标。。。

#include<iostream>
#include<map>
#include<cstring>
#include<vector>
using namespace std;
const int N=1e3+10;
int m,n;
int a[N],b[N];
int dp[N];
map<int,vector<int> >mp;
int main()
{
    cin>>m>>n;
    for(int i=1;i<=n;++i)
    {
        cin>>a[i]>>b[i];
        int c;
        cin>>c;
        mp[c].push_back(i);
    }
    map<int,vector<int> >::iterator  it;
    memset(dp,0,sizeof(dp));
    for(it=mp.begin();it!=mp.end();++it)
    {
        vector<int>tmp=it->second;
        for(int j=m;j>=0;--j)
        {
            for(int k=0;k<tmp.size();++k)
            {
                int i=tmp[k];           //i是本组中第k个物品的下标
                if(j-a[i]>=0)
                    dp[j]=max(dp[j],dp[j-a[i]]+b[i]);
            }
        }
    }
    cout<<dp[m]<<endl;
    return 0;
}

更好的方法是开一个数组记录每种的个数

二维数组，第一维表示种类，第二维存放物品下标_(:з」∠)_