mirror of
https://github.com/huihut/interview.git
synced 2024-03-22 13:10:48 +08:00
Update pack.cpp
Your PR can't be compiled, I modified some code. 1. Added the definition of T 2. Added delete heap variables 3. packIterative and printT separate two functions to uncouple code
This commit is contained in:
parent
e5b43e7632
commit
e7782962b7
|
@ -2,6 +2,8 @@
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
using namespace std;
|
using namespace std;
|
||||||
|
|
||||||
|
int **T = NULL; // 存储背包问题表格的数组指针
|
||||||
|
|
||||||
// 返回两个值的最大值
|
// 返回两个值的最大值
|
||||||
int max(int a, int b) {
|
int max(int a, int b) {
|
||||||
return (a > b) ? a : b;
|
return (a > b) ? a : b;
|
||||||
|
@ -9,26 +11,47 @@ int max(int a, int b) {
|
||||||
|
|
||||||
// 迭代法,能显示背包问题的表格
|
// 迭代法,能显示背包问题的表格
|
||||||
int packIterative(int n, int W, int *w, int *v) {
|
int packIterative(int n, int W, int *w, int *v) {
|
||||||
|
|
||||||
// 循环遍历n行
|
// 循环遍历n行
|
||||||
int T[i+1][j+1];
|
for (int i = 1; i <= n; ++i)
|
||||||
for (int i = 0; i <= n; ++i) {
|
{
|
||||||
// 循环遍历W列
|
// 循环遍历W列
|
||||||
for (int j = 0; j <= W; ++j) {
|
for (int j = 1; j <= W; ++j)
|
||||||
if (i == 0 || j == 0) {
|
{
|
||||||
T[i][j] = 0;
|
|
||||||
}
|
|
||||||
//第i个物品能装下,则比较包括第i个物品和不包括第i个物品,取其最大值
|
//第i个物品能装下,则比较包括第i个物品和不包括第i个物品,取其最大值
|
||||||
if ( w[i-1] <= j) {
|
if (w[i] <= j)
|
||||||
T[i][j] = max(v[i-1] + T[i - 1][j - w[i]], T[i - 1][j]);
|
T[i][j] = max(v[i] + T[i - 1][j - w[i]], T[i - 1][j]);
|
||||||
}
|
|
||||||
// 第i个物品不能装下,则递归装i-1个
|
// 第i个物品不能装下,则递归装i-1个
|
||||||
else {
|
else
|
||||||
T[i][j] = T[i - 1][j];
|
T[i][j] = T[i - 1][j];
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return T[n][W];
|
||||||
}
|
}
|
||||||
|
|
||||||
for (auto i = 0; i <= n; i++) {
|
// 递归法,不支持显示背包问题的表格
|
||||||
|
int packRecursive(int n, int W, int *w, int *v) {
|
||||||
|
// 结束条件(初始条件),i或者j为0时最大总价值为0
|
||||||
|
if (n == 0 || W == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
// 第i个物品不能装下,则递归装i-1个
|
||||||
|
if (w[n] > W) {
|
||||||
|
return packRecursive(n - 1, W, w, v);
|
||||||
|
}
|
||||||
|
//第i个物品能装下,则比较包括第i个物品和不包括第i个物品,取其最大值
|
||||||
|
else {
|
||||||
|
return max(v[n] + packRecursive(n - 1, W - w[n], w, v), packRecursive(n - 1, W, w, v));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 打印背包问题的表格
|
||||||
|
void printT(int n, int W)
|
||||||
|
{
|
||||||
|
// 打印n行
|
||||||
|
for (auto i = 0; i <= n; i++)
|
||||||
|
{
|
||||||
// 打印行数
|
// 打印行数
|
||||||
cout << i << ":\t";
|
cout << i << ":\t";
|
||||||
|
|
||||||
|
@ -41,26 +64,8 @@ int packIterative(int n, int W, int *w, int *v) {
|
||||||
// 换行
|
// 换行
|
||||||
cout << endl;
|
cout << endl;
|
||||||
}
|
}
|
||||||
return T[n][W];
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 递归法,不支持显示背包问题的表格
|
|
||||||
int packRecursive(int n, int W, int *w, int *v) {
|
|
||||||
// 结束条件(初始条件),i或者j为0时最大总价值为0
|
|
||||||
if (n == 0 || W == 0) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
// 第i个物品不能装下,则递归装i-1个
|
|
||||||
if (w[i] > W) {
|
|
||||||
return packRecursive(n - 1, W, w, v);
|
|
||||||
}
|
|
||||||
//第i个物品能装下,则比较包括第i个物品和不包括第i个物品,取其最大值
|
|
||||||
else {
|
|
||||||
return max(v[i] + packRecursive(n - 1, W - w[n], w , v), packRecursive(n - 1, W, w, v));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
int *w = NULL; // 存储每件物品重量的数组指针
|
int *w = NULL; // 存储每件物品重量的数组指针
|
||||||
int *v = NULL; // 存储每件物品价值的数组指针
|
int *v = NULL; // 存储每件物品价值的数组指针
|
||||||
|
@ -108,7 +113,7 @@ int main() {
|
||||||
// 输入背包的重量和价值
|
// 输入背包的重量和价值
|
||||||
for (auto i = 1; i <= n; i++)
|
for (auto i = 1; i <= n; i++)
|
||||||
{
|
{
|
||||||
cout << "请输入第 " << i << " 个背包的重量和价值(用空格隔开)" << endl;
|
cout << "请输入第 " << i << " 个物品的重量和价值(用空格隔开)" << endl;
|
||||||
cin >> w[i] >> v[i];
|
cin >> w[i] >> v[i];
|
||||||
if (cin.fail() || w[i] < 0 || v[i] < 0)
|
if (cin.fail() || w[i] < 0 || v[i] < 0)
|
||||||
{
|
{
|
||||||
|
@ -134,6 +139,8 @@ int main() {
|
||||||
{
|
{
|
||||||
// 迭代法,能显示背包问题的表格
|
// 迭代法,能显示背包问题的表格
|
||||||
cout << "能装下物品的最大价值为 " << packIterative(n, W, w, v) << endl;
|
cout << "能装下物品的最大价值为 " << packIterative(n, W, w, v) << endl;
|
||||||
|
cout << "------------------------------------------------" << endl;
|
||||||
|
printT(n, W);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
case 2:
|
case 2:
|
||||||
|
@ -151,6 +158,13 @@ int main() {
|
||||||
|
|
||||||
cout << "------------------------------------------------" << endl;
|
cout << "------------------------------------------------" << endl;
|
||||||
|
|
||||||
|
delete w;
|
||||||
|
delete v;
|
||||||
|
for (int i = 0; i <= n; ++i) {
|
||||||
|
delete[] T[i];
|
||||||
|
}
|
||||||
|
delete[] T;
|
||||||
|
|
||||||
system("pause");
|
system("pause");
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user