C++作业：将左右孩子互换 / 复制一棵二叉树

/* 邵发, 1309班,  */

/*
12周作业

练习1：编写递归算法、将左右孩子互换 （用递归）
练习2：编写复制一棵二叉树 （不用递归）

*/

#include <stdio.h>

// 定义节点
struct Node
{
int value;
Node* left, *right;
};

// 练习1：交换左右孩子节点
void SwapChildren(Node* tree)
{
if(!tree) return;

Node* t = tree->left;
tree->left = tree->right;
tree->right = t;

SwapChildren(tree->left);
SwapChildren(tree->right);
}

// 练习2：复制一棵二叉树

// 递归版本
// 也可以把引用改成**，不过可读性会差一些
void Copy_r(Node* oldtree, Node* &newtree)
{
// 空节点
if(oldtree == NULL)
{
newtree = NULL;
}
else
{
// 附加节点
newtree = new Node();
// 拷贝节点的值
newtree->value = oldtree->value;

// 递归拷贝子节点
Copy_r(oldtree->left, newtree->left);
Copy_r(oldtree->right,newtree->right);
}
}

// 非递归版本 （直接由递归版本改造, 由自定义栈替换递归逻辑）
void Copy_nr(Node* oldtree, Node** newtree)
{
Node*  stack_old[256];
Node** stack_new[256];  // 注意这里存放的是**
int top = 0; // 顶点位置，同时也是栈的元素个数

// 推入根节点
stack_old[top] = oldtree;
stack_new[top] = newtree;
top ++;

while(top > 0)
{
// 取节点复制
top--; // 先减再取

Node** pNew = stack_new[top];
Node*  pOld = stack_old[top];
//top --;

// 复制, 跟前面copy_r实际上相同
// 使用引用来去掉一个*，否则可读性欠佳
Node*& p = *pNew;
if(pOld == NULL)
{
p = NULL;
}
else
{
// 复制本节点
p = new Node();
p->value = pOld->value;

// 再复制子节点, 相当于递归调用
stack_new[top] = & (p->left);
stack_old[top] = pOld->left;
top ++;

stack_new[top] = & (p->right);
stack_old[top] = pOld->right;
top ++;
}
}
}