牛客多校第三场A（01背包变化）

牛客多校第三场A（01背包变化）

 

题意：

有n（1≤n≤36）个物品，每个物品有四种代价pi,ai,ci,mi，价值为gi（0≤pi,ai,ci,mi,gi≤36），

求四种代价分别不超过P,A,C,M（0≤P,A,C,M≤36）的条件下能获得的最大价值，输出所选择的物品。

分析：

01背包的思路，只是代价多了几个而已，数组开多几维就好了。

可以用vis[i][p][a][c][m]来表示在四种代价分别为p,a,c,m的状态下是否用第i件物品更新过，

从而实现记录方案的目的，详见代码。

 

 

#include<bits/stdc++.h>

using namespace std;

const int MAX = 38;

int dp[MAX][MAX][MAX][MAX];

bool path[MAX][MAX][MAX][MAX][MAX];

int p[MAX],a[MAX],c[MAX],m[MAX], g[MAX], cnt[MAX];

int N,P,A,C,M;

void solve(){

    memset(dp,0,sizeof(dp));

    memset(path,0,sizeof(path));

    //01背包的标准样式

    for(int i=0;i<N;++i){

        for(int j=P;j >= p[i];--j){

            for(int k=A;k >= a[i];--k){

                for(int l=C;l >= c[i];--l){

                    for(int o=M;o >= m[i];--o){

                        if(dp[j][k][l][o] < dp[j-p[i]][k-a[i]][l-c[i]][o-m[i]] + g[i]){

                            dp[j][k][l][o] = dp[j-p[i]][k-a[i]][l-c[i]][o-m[i]] + g[i];

                            path[i][j][k][l][o] = true;//如果被更新就标记这个情况为true

                        }

                    }

                }

            }

        }

    }

    //从最后的情况向前找就可以了。

    vector<int> vec;

    for(int i=N-1,j=P,k=A,l=C,o=M;i>=0&&j>=0&&k>=0&&l>=0&&o>=0;--i){

        if(path[i][j][k][l][o]){//表示这个物品被使用了。

            vec.push_back(i);

            j -= p[i];

            k -= a[i];

            l -= c[i];

            o -= m[i];

        }

    }

    cout << vec.size() << endl;

    reverse(vec.begin(),vec.end());

    for(vector<int>::iterator it = vec.begin();it != vec.end();++it){

        cout << *it << endl;

    }

}

int main(){

    ios::sync_with_stdio(false);

    cin.tie(0);

    cin >> N;

    for(int i=0;i<N;++i){

        cin >> p[i] >> a[i] >> c[i] >> m[i] >> g[i];

    }

    cin >> P >> A >> C >> M;

    solve();

    return 0;

}