做算法课设时候看到题目要求模拟函数递归时候栈的入栈出栈过程。本来想着直接调用系统递归函数即可,可是发现系统函数栈的空间非常小大约只有3000层,很容易爆栈。于是便有了用栈去模拟递归函数的想法,但是上网查了下貌似相关代码比较少,让GPT写的代码又牛头不对马嘴,于是手搓了一下。用栈模拟递归的过程,对递归过程,栈的理解都很有帮助。
二分递归函数模拟
第一个函数erfen为正常的二分递归函数,但是系统的递归栈最多能递归到3000层左右,太少了。我们可以用栈来模拟函数递归的过程,这样栈的层数可以达到我们设置的数组大小,我这里设置了二十万。
#include <iostream>
using namespace std;
const int N = 2e5 + 5;
struct binaryFrame {
int l, r;
int level;//记录当前层数
int step;//记录步数
}stackBinary[N];
/*需要模拟的二分递归函数
int erfen(int arr[], int l, int r, int key)
{
int ans = -1;
int mid = l + r >> 1;
if (arr[mid] == key && l == r) return r;
if (l >= r) return -1;
if (arr[mid] >= key) {
ans = erfen(arr, l, mid, key);
}
else {
ans = erfen(arr, mid + 1, r, key);
}
return ans;
}
*/
void binarySearch(int arr[], int l, int r, int key, int& index)
{
int nowLevel = 1, nowStep = 0, tt = 0, L = l, R = r;
tt++;
stackBinary[tt] = { l,r,nowLevel,nowStep };
cout << "压栈:" << "当前执行函数纸带范围为:(" << L << " , " << R << " )" << " 层数为" << nowLevel << " 执行步数为" << nowStep << "\n";
while (L < R)
{
int mid = L + R >> 1;
if (arr[mid] >= key)
{
R = mid;
stackBinary[++tt] = { L,R,++nowLevel,++nowStep };
cout << "压栈:" << "当前执行函数纸带范围为:(" << L << " , " << R << " )" << " 层数为" << nowLevel << " 执行步数为" << nowStep << "\n";
}
else {
L = mid + 1;
stackBinary[++tt] = { L,R,++nowLevel,++nowStep };
cout << "压栈:" << "当前执行函数纸带范围为:(" << L << " , " << R << " )" << " 层数为" << nowLevel << " 执行步数为" << nowStep << "\n";
}
}
if (arr[R] == key) {
index = R;
}
while (tt > 0)
{
cout << "出栈:" << "当前执行函数纸带范围为:(" << stackBinary[tt].l << " , " << stackBinary[tt].r << " )" << " 层数为" << stackBinary[tt].level << " 执行步数为" << stackBinary[tt].step << "\n";
tt--;
}
}
int main()
{
int arr[] = { 1,2,3,4,5,6,7,8,9,10 };
int n = 10;
int key = 2;
int index = -1;
binarySearch(arr, 0, n - 1, key, index);
if (index == -1) {
cout << "找不到这个吊数\n";
}
else {
cout << "这个B数在数组中的第" << index + 1 << "个位置" << "\n";
}
return 0;
}
显示压栈出栈过程
栈模拟回溯法解决01背包
#include<iostream>
#include<vector>
using namespace std;
typedef long long LL;
const int N = 2e5 + 5;
/*
// 全局变量用于存储最终结果
int max_value = 0;
vector<int> best_set;
/*
// 需要模拟的递归回溯函数
void knapsack(int index, int current_weight, int current_value, int capacity, const vector<int>& weights, const vector<int>& values, vector<int>& chosen_items) {
// 基础情况:如果考虑完所有物品
if (index == weights.size()) {
// 检查当前物品集合是否是一个有效解
if (current_weight <= capacity && current_value > max_value) {
max_value = current_value;
best_set = chosen_items;
}
return;
}
// 选择当前物品
chosen_items.push_back(index);
knapsack(index + 1, current_weight + weights[index], current_value + values[index], capacity, weights, values, chosen_items);
chosen_items.pop_back(); // 回溯
// 不选择当前物品
knapsack(index + 1, current_weight, current_value, capacity, weights, values, chosen_items);
}*/
struct backFrame
{
int index;
int current_weight;
int current_value;
int step;
vector<int> chosenItems;
int stage;
}stackBack[N];
void backBag(int n, int capacity, int& max_value, vector<int>& weights, vector<int>& values, vector<int>& chosenItems, vector<int>& best_set)
{
int tt = 0, nowStep = 0;
stackBack[++tt] = { 0,0,0,0,{},0 };
cout << "压栈:初始化压栈,压入 0 0 0 0 {} 0" << "\n";
while (tt > 0)
{
backFrame now = stackBack[tt];
cout << "当前执行函数层为:" << now.index << " " << now.current_weight << " " << now.current_value << " " << now.step << "\n";
if (stackBack[tt].index >= n) {
if (stackBack[tt].current_value > max_value && stackBack[tt].current_weight <= capacity)
{
max_value = stackBack[tt].current_value;
best_set = stackBack[tt].chosenItems;
cout << "更新best_set ";
for (auto i : best_set) cout << i + 1 << " ";
cout << "\n";
}
tt--;
cout << "出栈1:" << now.index << " " << now.current_weight<< " " << now.current_value << " " << now.step << "\n";
continue;
}
if (now.stage == 0) {//栈顶为初始状态的话直接下一层函数压栈
chosenItems.push_back(now.index);
stackBack[++tt] = { now.index + 1,now.current_weight + weights[now.index],now.current_value + values[now.index],now.step + 1,chosenItems,0 };
cout << "压栈1: " << now.index + 1 << " " << now.current_weight + weights[now.index] << " " << now.current_value + values[now.index] << " " << now.step + 1 << "\n";
stackBack[tt - 1].stage ++;
}
else if (now.stage == 1) {
chosenItems.pop_back();
stackBack[++tt] = { now.index + 1,now.current_weight,now.current_value,now.step + 1,chosenItems,0 };
cout << "压栈2: " << now.index + 1 << " " << now.current_weight << " " << now.current_value << " " << now.step + 1 << "\n";
stackBack[tt - 1].stage ++;
}
else if (now.stage == 2) {
tt--;
cout << "出栈2:" << now.index << " " << now.current_weight<< " " << now.current_value << " " << now.step << "\n";
}
}
}
int main()
{
vector<int> weights = { 1,2,3,5,7,6 };
vector<int> values = { 6,5,10,30,15,25 };
vector<int> chosenItems;
vector<int> best_set;
int capacity = 10, n = 6;
int max_value = -1;
/*
cout << "请输入物品件数和背包容量\n";
cin >> n >> capacity;
cout << "请输入" << n << "件物品,每件物品的质量,价值\n";
for (int i = 0; i < n; i++)
{
int x, y;
cin >> x >> y;
weights.push_back(x);
values.push_back(y);
}*/
backBag(n, capacity, max_value, weights, values, chosenItems, best_set);
LL sum = 0;
for (auto i : best_set)
{
cout << "选择了第" << i + 1 << "件物品\n";
sum += weights[i];
}
cout << "可以获得的最大价值为" << max_value << "\n";
cout << "所占用的背包空间为" << sum << "\n";
return 0;
}
/*
6 10
1 6
2 5
3 10
5 30
7 15
6 25
*/
栈模拟备忘录法解决01背包
#include <iostream>
#include <vector>
#include <cstring>
#include<Windows.h>
using namespace std;
const int N = 2e4 + 5;
int n, W;
int dp[505][1005]; // 备忘录数组
vector<int> weights;
vector<int> values;
/*需要模拟的备忘录递归函数
// 递归函数,包含备忘录算法逻辑,改为void类型
void knapsack(int idx, int W, int N, int& maxVal) {
// 基本情况
if (idx == N || W == 0) {
maxVal = 0;
cout << "边界:" << idx << " " << W << " " << maxVal << "\n";
return;
}
// 检查是否已经计算过
if (dp[idx][W] != -1) {
maxVal = dp[idx][W];
cout << "已经计算过:" << idx << " " << W << " " << maxVal << "\n";
return;
}
// 选择不包含当前物品
knapsack(idx + 1, W, N, maxVal);
int without = maxVal; // 从上一次递归中获取结果
cout << "without:" << idx << " " << W << " " << without << "\n";
int with = 0;
// 选择包含当前物品,如果背包容量足够
if (W >= weight[idx]) {
knapsack(idx + 1, W - weight[idx], N, with);
with = values[idx] + with; // 添加当前物品的价值
cout << "with:" << idx << " " << W << " " << with << "\n";
}
// 计算最大价值并存储备忘录结果
maxVal = max(without, with);
dp[idx][W] = maxVal; // 存储备忘录结果
//cout<<"idx=,W=,dp= "<<idx<<" "<<W<<" "<<dp[idx]
}*/
struct memoFrame
{
int index;
int with;
int without;
int max_value;
int capacity;
int stage;
}stackMemo[N];
void memoBag(int index, int capacity, vector<int>& weights, vector<int>& values)
{
int tt = 0;
stackMemo[++tt] = { 0,0,0,0,capacity,0 };
int max_value = 0;
while (tt > 0)
{
//cout << "cnt=" << cnt << "\n";
memoFrame& now = stackMemo[tt];
//cout << "当前函数层为,index,capacity,stage=" << now.index << " " << now.capacity << " " << now.stage << "\n";
if ((now.index == n || now.capacity <= 0)) {
memoFrame& last = stackMemo[tt - 1];
last.max_value = 0;
//cout << "当前层为:,返回给上一层的返回值为" << now.index << " " << now.capacity << " " << last.max_value << "\n";
tt--;
//cout << "出栈:index,capacity,return" << now.index << " " << now.capacity << " " << last.max_value << "\n";
continue;
}
if (dp[now.index][now.capacity] != -1)
{
memoFrame& last = stackMemo[tt - 1];
last.max_value = dp[now.index][now.capacity];
//cout << "当前层为:,返回给上一层的返回值为" << now.index << " " << now.capacity << " " << last.max_value << "\n";
tt--;
//cout << "出栈:index,capacity,return" << now.index << " " << now.capacity << " " << last.max_value << "\n";
continue;
}
if (now.stage == 0)
{
stackMemo[++tt] = { now.index + 1,0,0,0,now.capacity,0 };
//cout << "压栈0:" << now.index + 1 << " " << now.capacity << " " << "\n";
now.stage++;
}
else if (now.stage == 1)
{
now.without = now.max_value;
//cout << "更新without:index, without:" << now.index << " " << now.without << "\n";
now.stage++;
}
else if (now.stage == 2)
{
if (now.capacity >= weights[now.index]) stackMemo[++tt] = { now.index + 1, 0, 0, 0, now.capacity - weights[now.index], 0 };
// cout << "压栈1:" << now.index + 1 << " " << now.capacity << " " << "\n";
now.stage++;
}
else if (now.stage == 3)
{
if (now.capacity >= weights[now.index])
{
now.with = now.max_value + values[now.index];
//cout << "更新with, index,with:" << now.index << " " << now.with << "\n";
}
now.stage++;
}
else if (now.stage == 4)
{
memoFrame& last = stackMemo[tt - 1];
dp[now.index][now.capacity] = max(now.with, now.without);
for (int i = 0; i < n; i++)
{
for (int j = 0; j < W; j++)
{
cout << dp[i][j] << " ";
}
cout << "\n";
}
last.max_value = dp[now.index][now.capacity];
Sleep(1000);
//cout << "更新dp: index,dp:" << now.index << " " << last.max_value << "\n";
tt--;
//cout << "出栈:index,capacity,return" << now.index << " " << now.capacity << " " << last.max_value << "\n";
}
}
}
int main() {
cin >> n >> W;; // 读取物品数量和背包容量
weights.resize(n);
values.resize(n);
for (int i = 0; i < n; ++i) {
cin >> weights[i] >> values[i];
}
memset(dp, -1, sizeof(dp));
memoBag(0, W, weights, values);
cout << "最大价值:" << dp[0][W] << endl;
return 0;
}
/*
6 10
1 6
2 5
3 10
5 30
7 15
6 25
*/
把三个题整合到一块
#include <iostream>
#include <vector>
#include <cstring>
#include<iomanip>
#include <windows.h>
using namespace std;
const int N = 2e4 + 5;
typedef long long LL;
int n, W;//n可以是二分的数组长度,也可以是背包的物品个数,W是背包容量
int ansPos=-1;//二分搜索到的答案位置
int arr[N];
int dp[505][1005]; // 备忘录数组
vector<int> weights;
vector<int> values;
struct binaryFrame {
int l, r;
int level;//记录当前层数
int step;//记录步数
}stackBinary[N];
struct backFrame
{
int index;
int current_weight;
int current_value;
int step;
vector<int> chosenItems;
int stage;
}stackBack[N];
struct memoFrame
{
int index;
int with;
int without;
int max_value;
int capacity;
int stage;
}stackMemo[N];
void binarySearch(int l, int r, int key)
{
int nowLevel = 1, nowStep = 0, tt = 0, L = l, R = r;
tt++;
stackBinary[tt] = { l,r,nowLevel,nowStep };
cout<<right<<setw(50) << "压栈:" << "当前执行函数纸带范围为:(" << L << " , " << R << " )" << " 层数为" << nowLevel << " 执行步数为" << nowStep << "\n";
Sleep(200);
while (L < R)
{
int mid = L + R >> 1;
if (arr[mid] >= key)
{
R = mid;
stackBinary[++tt] = { L,R,++nowLevel,++nowStep };
cout<<right<<setw(50) << "压栈:" << "当前执行函数纸带范围为:(" << L << " , " << R << " )" << " 层数为" << nowLevel << " 执行步数为" << nowStep << "\n";
Sleep(200);
}
else {
L = mid + 1;
stackBinary[++tt] = { L,R,++nowLevel,++nowStep };
cout<<right<<setw(50) << "压栈:" << "当前执行函数纸带范围为:(" << L << " , " << R << " )" << " 层数为" << nowLevel << " 执行步数为" << nowStep << "\n";
Sleep(200);
}
}
if (arr[R] == key) {
ansPos = R;
}
while (tt > 0)
{
cout<<right<<setw(50) << "出栈:" << "当前执行函数纸带范围为:(" << stackBinary[tt].l << " , " << stackBinary[tt].r << " )" << " 层数为" << stackBinary[tt].level << " 执行步数为" << stackBinary[tt].step << "\n";
Sleep(200);
tt--;
}
}
void backBag(int n, int capacity, int& max_value, vector<int>& weights, vector<int>& values, vector<int>& chosenItems, vector<int>& best_set)
{
int tt = 0, nowStep = 0;
stackBack[++tt] = { 0,0,0,0,{},0 };
cout<<right<<setw(50) << "压栈:初始化压栈,压入 0 0 0 0 {} 0" << "\n";
Sleep(200);
while (tt > 0)
{
backFrame now = stackBack[tt];
cout << right << setw(50) << "当前执行函数层为:下标为:" << now.index << " 当前重量:" << now.current_weight << " 当前价值:" << now.current_value << " 当前步数:" << now.step << "\n";
Sleep(200);
if (stackBack[tt].index >= n) {
if (stackBack[tt].current_value > max_value && stackBack[tt].current_weight <= capacity)
{
max_value = stackBack[tt].current_value;
best_set = stackBack[tt].chosenItems;
}
tt--;
cout<<right<<setw(50) << "出栈1:出栈下标:" << now.index << " 当前重量:" << now.current_weight << " 当前价值:" << now.current_value << " 当前步数:" << now.step << "\n";
Sleep(200);
continue;
}
if (now.stage == 0) {//栈顶为初始状态的话直接下一层函数压栈
chosenItems.push_back(now.index);
stackBack[++tt] = { now.index + 1,now.current_weight + weights[now.index],now.current_value + values[now.index],now.step + 1,chosenItems,0 };
cout<<right<<setw(50) << "压栈1: 压栈下标:" << now.index + 1 << " 压栈重量:" << now.current_weight + weights[now.index] << " 压栈价值:" << now.current_value + values[now.index] << " 压栈步数:" << now.step + 1 << "\n";
Sleep(200);
stackBack[tt - 1].stage++;
}
else if (now.stage == 1) {
chosenItems.pop_back();
stackBack[++tt] = { now.index + 1,now.current_weight,now.current_value,now.step + 1,chosenItems,0 };
cout<<right<<setw(50) << "压栈2: 压栈下标:" << now.index + 1 << " 压栈重量:" << now.current_weight << " " << now.current_value << " " << now.step + 1 << "\n";
Sleep(200);
stackBack[tt - 1].stage++;
}
else if (now.stage == 2) {
tt--;
cout<<right<<setw(50) << "出栈2:" << now.index << " 出栈重量:" << now.current_weight << " 出栈价值:" << now.current_value << " 出栈步数" << now.step << "\n";
Sleep(200);
}
}
}
void memoBag(int index, int capacity, vector<int>& weights, vector<int>& values)
{
int tt = 0;
stackMemo[++tt] = { 0,0,0,0,capacity,0 };
int max_value = 0;
while (tt > 0)
{
memoFrame& now = stackMemo[tt];
cout<<right<<setw(50) << "当前函数层为,当前执行下标:" << now.index << " 当前背包剩余容量:" << now.capacity << " 当前stage:" << now.stage << "\n";
Sleep(200);
if ((now.index == n || now.capacity <= 0)) {
memoFrame& last = stackMemo[tt - 1];
last.max_value = 0;
//cout << "当前层为:,返回给上一层的返回值为" << now.index << " " << now.capacity << " " << last.max_value << "\n";
tt--;
cout<<right<<setw(50) << "出栈:出栈下标:" << now.index << " 出栈时剩余背包容量:" << now.capacity << " 出栈返回值:" << last.max_value << "\n";
Sleep(200);
continue;
}
if (dp[now.index][now.capacity] != -1)
{
memoFrame& last = stackMemo[tt - 1];
last.max_value = dp[now.index][now.capacity];
//cout << "当前层为:,返回给上一层的返回值为" << now.index << " " << now.capacity << " " << last.max_value << "\n";
tt--;
cout<<right<<setw(50) << "出栈:出栈下标:" << now.index << " 出栈时剩余背包容量:" << now.capacity << " 出栈时返回值:" << last.max_value << "\n";
Sleep(200);
continue;
}
if (now.stage == 0)
{
stackMemo[++tt] = { now.index + 1,0,0,0,now.capacity,0 };
cout<<right<<setw(50) << "压栈0:压栈下标:" << now.index + 1 << " 压栈背包容量;" << now.capacity << " " << "\n";
Sleep(200);
now.stage++;
}
else if (now.stage == 1)
{
now.without = now.max_value;
//cout << "更新without:index, without:" << now.index << " " << now.without << "\n";
now.stage++;
}
else if (now.stage == 2)
{
if (now.capacity >= weights[now.index]) stackMemo[++tt] = { now.index + 1, 0, 0, 0, now.capacity - weights[now.index], 0 };
cout<<right<<setw(50) << "压栈1:压栈下标:" << now.index + 1 << " 压栈背包容量:" << now.capacity-weights[now.index] << " " << "\n";
Sleep(200);
now.stage++;
}
else if (now.stage == 3)
{
if (now.capacity >= weights[now.index])
{
now.with = now.max_value + values[now.index];
//cout << "更新with, index,with:" << now.index << " " << now.with << "\n";
}
now.stage++;
}
else if (now.stage == 4)
{
memoFrame& last = stackMemo[tt - 1];
dp[now.index][now.capacity] = max(now.with, now.without);
last.max_value = dp[now.index][now.capacity];
//cout << "更新dp: index,dp:" << now.index << " " << last.max_value << "\n";
tt--;
cout<<right<<setw(50) << "出栈:出栈下标:" << now.index << " 出栈时背包容量:" << now.capacity << " 这一层出栈的返回值:" << last.max_value << "\n";
Sleep(200);
}
}
}
int main()
{
int op;
cout<<right<<setw(50) << "请选择要执行的递归函数模拟:1.二分搜索,2.回溯法,3.备忘录法\n";
while(cin >> op) {
if (op == 1) {
int key;
cout << right << setw(50) << "栈模拟递归二分算法\n\n\n";
cout << right << setw(50) << "请输入查询数组长度\n";
cin >> n;
cout << right << setw(50) << "请输入" << n << "个数字\n";
for (int i = 0; i < n; i++)
{
cin >> arr[i];
}
cout << right << setw(50) << "请输入需要查找的数字key\n";
cin >> key;
binarySearch(0, n - 1, key);
if (ansPos == -1) cout << right << setw(50) << "找不到这个吊数\n";
else cout << right << setw(50) << "数字" << key << "在数组中的第" << ansPos + 1 << "个位置\n";
}
else if (op == 2)
{
weights.clear();
values.clear();
cout << right << setw(50) << "栈模拟递归回溯算法\n\n\n";
cout << right << setw(50) << "请输入物品件数和背包容量\n";
cin >> n >> W;
cout << right << setw(50) << "请输入" << n << "件物品的质量和价值\n";
for (int i = 0; i < n; i++)
{
int x, y;
cin >> x >> y;
weights.push_back(x);
values.push_back(y);
}
int max_value = -1;
vector<int> best_set;
vector<int> chosenItems;
backBag(n, W, max_value, weights, values, chosenItems, best_set);
cout << right << setw(50) << "可以获得的最大价值为:" << max_value << "\n";
for (auto i : best_set)
{
cout << right << setw(50) << "选择了第" << i + 1 << "件物品\n";
}
}
else if (op == 3)
{
weights.clear();
values.clear();
cout << right << setw(50) << "栈模拟递归备忘录算法\n\n\n";
cout << right << setw(50) << "请输入物品件数和背包容量\n";
cin >> n >> W;
cout << right << setw(50) << "请输入" << n << "件物品的质量和价值\n";
for (int i = 0; i < n; i++)
{
int x, y;
cin >> x >> y;
weights.push_back(x);
values.push_back(y);
}
memset(dp, -1, sizeof(dp));
memoBag(0, W, weights, values);
cout << right << setw(50) << "能获得的最大价值为:" << dp[0][W] << "\n";
}
cout << right << setw(50) << "1.继续,2.结束\n";
cin >> op;
if (op == 2) break;
cout << right << setw(50) << "请选择要执行的递归函数模拟:1.二分搜索,2.回溯法,3.备忘录法\n";
}
return 0;
}
/*二分搜索样例输入
10
1 2 3 4 5 6 7 8 9 10
2
*/
/*回溯与备忘录样例输入
6 10
1 6
2 5
3 10
5 30
7 15
6 25
*/
/*需要模拟的二分递归函数
int erfen(int arr[], int l, int r, int key)
{
int ans = -1;
int mid = l + r >> 1;
if (arr[mid] == key && l == r) return r;
if (l >= r) return -1;
if (arr[mid] >= key) {
ans = erfen(arr, l, mid, key);
}
else {
ans = erfen(arr, mid + 1, r, key);
}
return ans;
}
*/
/*
// 需要模拟的递归回溯函数
void knapsack(int index, int current_weight, int current_value, int capacity, const vector<int>& weights, const vector<int>& values, vector<int>& chosen_items) {
// 基础情况:如果考虑完所有物品
if (index == weights.size()) {
// 检查当前物品集合是否是一个有效解
if (current_weight <= capacity && current_value > max_value) {
max_value = current_value;
best_set = chosen_items;
}
return;
}
// 选择当前物品
chosen_items.push_back(index);
knapsack(index + 1, current_weight + weights[index], current_value + values[index], capacity, weights, values, chosen_items);
chosen_items.pop_back(); // 回溯
// 不选择当前物品
knapsack(index + 1, current_weight, current_value, capacity, weights, values, chosen_items);
}*/
/*需要模拟的备忘录递归函数
// 递归函数,包含备忘录算法逻辑,改为void类型
void knapsack(int idx, int W, int N, int& maxVal) {
// 基本情况
if (idx == N || W == 0) {
maxVal = 0;
cout << "边界:" << idx << " " << W << " " << maxVal << "\n";
return;
}
// 检查是否已经计算过
if (dp[idx][W] != -1) {
maxVal = dp[idx][W];
cout << "已经计算过:" << idx << " " << W << " " << maxVal << "\n";
return;
}
// 选择不包含当前物品
knapsack(idx + 1, W, N, maxVal);
int without = maxVal; // 从上一次递归中获取结果
cout << "without:" << idx << " " << W << " " << without << "\n";
int with = 0;
// 选择包含当前物品,如果背包容量足够
if (W >= weight[idx]) {
knapsack(idx + 1, W - weight[idx], N, with);
with = values[idx] + with; // 添加当前物品的价值
cout << "with:" << idx << " " << W << " " << with << "\n";
}
// 计算最大价值并存储备忘录结果
maxVal = max(without, with);
dp[idx][W] = maxVal; // 存储备忘录结果
//cout<<"idx=,W=,dp= "<<idx<<" "<<W<<" "<<dp[idx]
}*/