#include #include #include using namespace std; /* 0-1 ナップサック問題:全探索 */ int knapsackDFS(vector &wgt, vector &val, int i, int c) { // すべての品物を選び終えたか、ナップサックの残り容量がなければ、価値 0 を返す if (i == 0 || c == 0) { return 0; } // ナップサックの容量を超える場合、入れない選択しかできない if (wgt[i - 1] > c) { return knapsackDFS(wgt, val, i - 1, c); } // 品物 i を入れない場合と入れる場合の最大価値を計算する int no = knapsackDFS(wgt, val, i - 1, c); int yes = knapsackDFS(wgt, val, i - 1, c - wgt[i - 1]) + val[i - 1]; // 2 通りの選択のうち、価値が大きいほうを返す return max(no, yes); } /* 0-1 ナップサック問題:メモ化探索 */ int knapsackDFSMem(vector &wgt, vector &val, vector> &mem, int i, int c) { // すべての品物を選び終えたか、ナップサックの残り容量がなければ、価値 0 を返す if (i == 0 || c == 0) { return 0; } // 記録済みなら、その値をそのまま返す if (mem[i][c] != -1) { return mem[i][c]; } // ナップサックの容量を超える場合、入れない選択しかできない if (wgt[i - 1] > c) { return knapsackDFSMem(wgt, val, mem, i - 1, c); } // 品物 i を入れない場合と入れる場合の最大価値を計算する int no = knapsackDFSMem(wgt, val, mem, i - 1, c); int yes = knapsackDFSMem(wgt, val, mem, i - 1, c - wgt[i - 1]) + val[i - 1]; // 2 通りの選択のうち、価値が大きいほうを記録して返す mem[i][c] = max(no, yes); return mem[i][c]; } /* 0-1 ナップサック問題:動的計画法 */ int knapsackDP(vector &wgt, vector &val, int cap) { int n = wgt.size(); // dp テーブルを初期化する vector> dp(n + 1, vector(cap + 1, 0)); // 状態遷移 for (int i = 1; i <= n; i++) { for (int c = 1; c <= cap; c++) { if (wgt[i - 1] > c) { // ナップサックの容量を超える場合、品物 i を選ばない dp[i][c] = dp[i - 1][c]; } else { // 品物 i を選ばない場合と選ぶ場合の大きいほうの値 dp[i][c] = max(dp[i - 1][c], dp[i - 1][c - wgt[i - 1]] + val[i - 1]); } } } return dp[n][cap]; } /* 0-1 ナップサック問題:空間を最適化した動的計画法 */ int knapsackDPComp(vector &wgt, vector &val, int cap) { int n = wgt.size(); // dp テーブルを初期化する vector dp(cap + 1, 0); // 状態遷移 for (int i = 1; i <= n; i++) { // 逆順に走査する for (int c = cap; c >= 1; c--) { if (wgt[i - 1] <= c) { // 品物 i を選ばない場合と選ぶ場合の大きいほうの値 dp[c] = max(dp[c], dp[c - wgt[i - 1]] + val[i - 1]); } } } return dp[cap]; } /* 実行用コード */ int main() { vector wgt = {10, 20, 30, 40, 50}; vector val = {50, 120, 150, 210, 240}; int cap = 50; int n = wgt.size(); // 全探索 int res = knapsackDFS(wgt, val, n, cap); cout << "ナップサックの容量を超えない品物の最大価値は " << res << endl; // メモ化探索 vector> mem(n + 1, vector(cap + 1, -1)); res = knapsackDFSMem(wgt, val, mem, n, cap); cout << "ナップサックの容量を超えない品物の最大価値は " << res << endl; // 動的計画法 res = knapsackDP(wgt, val, cap); cout << "ナップサックの容量を超えない品物の最大価値は " << res << endl; // 空間を最適化した動的計画法 res = knapsackDPComp(wgt, val, cap); cout << "ナップサックの容量を超えない品物の最大価値は " << res << endl; return 0; }