在回溯法①中介绍了八皇后问题,接着老师让我们试着用回溯法解决一下01背包问题(话说01背包问题是不是已经有好多解法了,动态规划,贪心,回溯........也是万金油题目了),并预测期末可能会出一道大题。那就试着写一下吧。题目链接:01背包问题。
看一下题目:
相比于八皇后问题,感觉这个问题的约束条件会简单很多,就是一件物品选或者不选,如果选了就要考虑它有没有超过背包的重量即可,把每种可能的情况与当前存储的最大值进行比较,再进行比较即可(但是这个方法只能通过部分的数据,可能时间复杂度较大,最合适的方法还是动态规划,这里只是作为一个回溯法的使用练习一下)。
#include<iostream>
using namespace std;
const int c = 1024;
int v[c], w[c],col[c];//v代表价值,w代表容量,col代表当前的选择
int answer;//用来保存最大的答案
bool Condition(int k,int V) {
int temp = 0;
for (int i = 1; i <= k; i++) {
if (col[i] == 1) {
temp += w[i];
}
}
if (temp > V) { return false; }
return true;
}
void Trace(int N,int V) {
int k = 1;
for (int i = 1; i <= N; i++) {
col[k] = -1;
}//此处由于选和不选的值是从0到1,所以初始化的时候要从-1开始。
while (k > 0) {
col[k] = col[k]+1;//这里代表着迭代更新搜索,初始从0开始,然后是1,超过1则剪枝
while (col[k] <= 1 && !Condition(k, V)) {
col[k] += 1;
}
if (col[k] <= 1) {
if (k == N) {
int temp = 0;
for (int i = 1; i <= N; i++) {
if (col[i]==1) { temp += v[i]; }
}
if (temp > answer) { answer = temp; }
}
else { k = k + 1; col[k] = -1; }
}
else { k = k - 1; }
}
}
int main() {
int N, V;
cin >> N >> V;
for (int i = 1; i <= N; i++) {
cin >> w[i] >> v[i];
}
Trace(N, V);
cout << answer << endl;
}
我看网上有很多方法是用dfs(深度优先方法)来做的,这里我这是根据老师对八皇后问题的方法进行稍微改变一下,感觉跟dfs有点差别,但是可能更好理解一点。有兴趣也可以自行寻找dfs的做法~