P2732 [USACO3.3]商店购物 Shopping Offers

原题链接

考察:背包dp

思路:

        做之前看成01背包,然后答案一直不对,醉了.

        实际是完全背包,因为需要的东西不超过5类,且不超过5个,所以可以用5维数组.5类物品的数目都是体积.

        需要将编号离散化,这里用的是map映射.

        但是注意的是我的本地IDE 在mp[x] = mp.size()是从1开始,但是洛谷的IDE是从0开始.可能不同编辑器不一样,还是老老实实设置一个变量较好.

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <vector>
using namespace std;
const int N = 105,M = 6,INF = 0x3f3f3f3f;
typedef pair<int,int> PII;
vector<PII> dis[N];
map<int,int> mp;
int s,n,price[N],f[M][M][M][M][M],cnt;
int need[M];
struct LIST{
    int nums[6],pce;
}list[N+M];
int get(int x)
{
    if(!mp.count(x)) mp[x] =mp.size();
    return mp[x];
}
void init()
{
    scanf("%d",&s);
    for(int i=1;i<=s;i++)//折扣 
    {
        int m,c,k;
        scanf("%d",&m);
        for(int j=1;j<=m;j++)
        {
            scanf("%d%d",&c,&k);
            dis[i].push_back({c,k});
        }
        scanf("%d",&price[i]);
    }
    scanf("%d",&n);
    for(int i=1;i<=n;i++)//需求 
    {
        int c,k,p; scanf("%d%d%d",&c,&k,&p);
        list[++cnt].nums[get(c)] = 1;
        list[cnt].pce = p;
        need[get(c)] = k;
    }
    cnt++;
    for(int i=1;i<=s;i++,cnt++)
        for(int j=0;j<dis[i].size();j++)
        {
            int x = get(dis[i][j].first),y = dis[i][j].second;
            if(x>5) continue;
            list[cnt].nums[x] = y;
            list[cnt].pce = price[i];
        }
}
void work()
{
    memset(f,0x3f,sizeof f);
    f[0][0][0][0][0] = 0;
    for(int I=1;I<=cnt;I++)
     for(int i=list[I].nums[1];i<=need[1];i++)
       for(int j=list[I].nums[2];j<=need[2];j++)
         for(int k=list[I].nums[3];k<=need[3];k++)
          for(int p=list[I].nums[4];p<=need[4];p++)
           for(int l=list[I].nums[5];l<=need[5];l++)
           {
                 int a = list[I].nums[1],b = list[I].nums[2];
                 int c = list[I].nums[3],d = list[I].nums[4];
                 int e = list[I].nums[5],g = f[i][j][k][p][l];
                 if(f[i-a][j-b][k-c][p-d][l-e]>=INF) continue;
                 f[i][j][k][p][l] = min(f[i-a][j-b][k-c][p-d][l-e]+list[I].pce,g);
           }
            
}
int main()
{
    init();
    work();
    printf("%d\n",f[need[1]][need[2]][need[3]][need[4]][need[5]]);
    return 0;
}