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:
huihut 2019-09-12 02:09:58 +08:00 committed by GitHub
parent e5b43e7632
commit e7782962b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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; // 存储每件物品价值的数组指针
@ -94,8 +99,8 @@ int main() {
// 分配空间 // 分配空间
// 对w和v分配n+1大小 // 对w和v分配n+1大小
w = new int[n+1]; w = new int[n + 1];
v = new int[n+1]; v = new int[n + 1];
// 对T分配n+1行并初始化为0 // 对T分配n+1行并初始化为0
T = new int *[n + 1](); T = new int *[n + 1]();
@ -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;
} }