朴素版本:
#include<bits/stdc++.h> using namespace std; const int maxn=1005; int dp[maxn][maxn]; int v[maxn][maxn],w[maxn][maxn],s[maxn]; int main(){ int n,m;scanf("%d%d",&n,&m); for(int i=1;i<=n;i++){ scanf("%d",&s[i]); for(int j=1;j<=s[i];j++){ scanf("%d%d",&v[i][j],&w[i][j]); } } for(int i=1;i<=n;i++){ for(int j=0;j<=m;j++){ dp[i][j]=dp[i-1][j]; for(int k=1;k<=s[i];k++){ if(j>=v[i][k]) dp[i][j]=max(dp[i][j],dp[i-1][j-v[i][k]]+w[i][k]); } } } printf("%d ",dp[n][m]); }
一维数组优化版本(类01背包):
#include<bits/stdc++.h> using namespace std; const int maxn=1005; int dp[maxn]; int v[maxn][maxn],w[maxn][maxn],s[maxn]; int main(){ int n,m;scanf("%d%d",&n,&m); for(int i=1;i<=n;i++){ scanf("%d",&s[i]); for(int j=1;j<=s[i];j++){ scanf("%d%d",&v[i][j],&w[i][j]); } } for(int i=1;i<=n;i++){ for(int j=m;j>=0;j--){ for(int k=1;k<=s[i];k++){ if(j>=v[i][k]) dp[j]=max(dp[j],dp[j-v[i][k]]+w[i][k]); } } } printf("%d ",dp[m]); }