洛谷 P1507 NASA的食物计划 (dp 01背包问题)

题目背景

NASA（美国航空航天局）因为航天飞机的隔热瓦等其他安全技术问题一直大伤脑筋，因此在各方压力下终止了航天飞机的历史，但是此类事情会不会在以后发生，谁也无法保证。所以，在遇到这类航天问题时，也许只能让航天员出仓维修。但是过多的维修会消耗航天员大量的能量，因此 NASA 便想设计一种食品方案，使体积和承重有限的条件下多装载一些高卡路里的食物。

题目描述

航天飞机的体积有限，当然如果载过重的物品，燃料会浪费很多钱，每件食品都有各自的体积、质量以及所含卡路里。在告诉你体积和质量的最大值的情况下，请输出能达到的食品方案所含卡路里的最大值，当然每个食品只能使用一次。

输入格式

第一行 2 个整数，分别代表体积最大值 h 和质量最大值 t。

第二行 1 个整数代表食品总数 n。

接下来 n 行每行 3 个数 体积 hi​，质量 ti​，所含卡路里 ki​。

输出格式

一个数，表示所能达到的最大卡路里（int 范围内）

输入输出样例

输入 #1

320 350
4
160 40 120
80 110 240
220 70 310
40 400 220

输出 #1

550

说明/提示

对于 100%100% 的数据，ℎ,𝑡,ℎ𝑖,𝑡𝑖≤400，𝑛≤50，𝑘𝑖≤500。

01背包问题是动态规划算法中很经典的一个模型，这个模型的建模过程就是找属性和集合，那很明显这道题的属性就是卡路里，集合就是物品。

那就是要么选这个物品，要么不选这个物品状态转移就是：dp[i][j][k] = max(dp[i-1][j-w[i].h][k-w[i].t])

代码：

#include <bits/stdc++.h>
using namespace std;

const int N = 505;
int dp[55][N][N];

struct L{
    int h;
    int t;
    int v;
}w[55];

int H,T,m;

int main()
{
    cin >> H >> T;
    cin >> m;
    for(int i=1;i<=m;i++){
        cin >> w[i].h >> w[i].t >> w[i].v;
        // cout << w[i].h << " " << endl;
    }
    for(int i=1;i<=m;i++){
        for(int j=0;j<=H;j++){
            for(int k=0;k<=T;k++){
                dp[i][j][k] = dp[i-1][j][k];
                if(j >= w[i].h && k >= w[i].t) dp[i][j][k] = max(dp[i][j][k],dp[i-1][j-w[i].h][k-w[i].t] + w[i].v);
            }
        }
    }
    cout << dp[m][H][T] << endl;
    return 0;
}

加油