backpack 3 (lintcode 440)
2019-02-12 本文已影响0人
Ariana不会哭
图片.png
图片.png
图片.png
class Solution {
public:
int backPackIII( vector<int> &A, vector<int> &V,int m) {
int ss = A.size();
if (ss == 0)
return 0;
vector<vector<int>> dp(ss + 1, vector<int>(m + 1, 0));
for (int j = 1; j <= m; j++) {
dp[0][j] = -1;
}
for (int i = 1; i <= ss; i++) {
for (int j = 0; j <= m; j++) {
dp[i][j] = dp[i - 1][j];
if (j - A[i - 1] >= 0 && dp[i][j - A[i - 1]] != -1) {
dp[i][j] = max(dp[i][j], dp[i][j - A[i - 1]] + V[i - 1]);
}
}
}
int ans = 0;
for (int i = m; i >= 0; i--) {
ans = max(ans, dp[ss][i]);
}
return ans;
}
};