「牛客dp专题1」队伍配置

date:2022/1/12
URL:https://ac.nowcoder.com/acm/contest/24213/1025

题目描述

n个物品 和 m 个装饰品，每个物品和装饰品有价值和体积，装饰品无法单独取，必须依附于物品，且物品最多取5个，物品和装饰品均只能取一次，一个物品最多装饰一个饰品，问不超过体积v，可获取的最大价值。

思路

一眼01背包，然后没思路
题目限制比较多，这里我们考虑状态$f[i][j][k]$ 表示考虑到体积不超过v时，取j件物品和k件装饰品的最大价值 , j >= k
这里由于装饰品是依赖于物品的，所以我们可以先更新出不取装饰品时各各状态的最大价值
状态转移方程为：$f[j][k][0]=max(f[j][k][0],f[j-v][k-1][0]+w)$
后再更新取装饰品的各各状态的最大价值，状态更新过程中取max即可
状态转移方程为：$f[j][k][l]=max(f[j][k][l],f[j-v][k][l-1]+w)$

Code

/*
    author:@bzdhxs
    date:2022/1/12
    URL:https://ac.nowcoder.com/acm/contest/24213/1025
    限制较多的01背包
    /*运行时间4ms,占用内存416KB*/
*/
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<queue>
using namespace std;
template <typename T>
inline void read(T &s){s = 0;T w = 1, ch = getchar();while (!isdigit(ch)) { if (ch == '-') w = -1; ch = getchar(); }while (isdigit(ch))  { s = (s << 1) + (s << 3) + (ch ^48); ch = getchar();} s *= w;}
template <typename T>
inline void write(T s){if (s < 0) putchar('-'), s = -s;if (s > 9) write(s / 10);putchar(s % 10 + '0');}
#define int long long
#define _orz ios::sync_with_stdio(false),cin.tie(0)
#define mem(str,num) memset(str,num,sizeof(str))
#define forr(i,a,b) for(int i = a; i <= b;i++)
#define forn(i,n) for(int i = 0; i < n; i++)
#define dbg() cout <<"0k!"<< endl;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int N = 1e6+10;
int n,m,d;

int f[310][6][6]; //  考虑到 (i,j,k) 花费小于等于i时, 从者的数量为j,衣服数为k时的最大价值; 
signed main()
{
    cin >> n >> m >> d;
    // 处理出衣服个数为0时的最大价值;
    int res = -inf;
    forr(i,1,n){
        int x,y;cin >> x >> y;

        for(int j = d; j >= y;j--){
            for(int k = 1;k <= 5;k++){// 从者i:当第k个行者取;
                f[j][k][0] = max(f[j][k][0],f[j-y][k-1][0]+x);
                res = max(res,f[j][k][0]);
            }
        }
    }

    forr(i,1,m){
        int x,y;cin >> x >> y;
        for(int j = d ; j >= y;j--){
            for(int k = 1; k <= 5;k++)
                for(int l = 1; l <= k;l++){ // 这里状态转移考虑到的是这件装饰品是当做第l个取的
                    f[j][k][l] = max(f[j][k][l],f[j-y][k][l-1]+x);// 衣服i:当第l件去取
                    res = max(res,f[j][k][l]);
                }
        }
    }
    cout << res << endl;
    return 0;
}