复制二叉树

递归

/*
复制一棵二叉树的非递归算法
*/
#include <iostream>
#include <deque>
#include <malloc.h>
using namespace std;
typedef struct BTNode {
    int data;
    struct BTNode *left, *right;
}*BiTree;

BiTree creat_bt(){ //按扩展前序建二叉树
    BiTree t;
	int x;
    cin >> x; 
    if (x==0) t=NULL;
    else 
    { 
		t = (BiTree)malloc(sizeof(BTNode));
        t->data = x;
        t->left = creat_bt();
        t->right = creat_bt();
    }
    return t;
}
//递归复制，不分配空间 
void copyTree(BiTree &T, BiTree p){
	T = p;
	if(p == NULL)
		return;
		
	copyTree(T->left, p->left);
	copyTree(T->right, p->right);
}
//递归复制，分配空间
void copyTree1(BiTree &T, BiTree p){
	if(p){
		T = (BiTree)malloc(sizeof(BTNode));
		T->data = p->data;
	}else{
		p = NULL;
		return;
	}
		
	copyTree(T->left, p->left);
	copyTree(T->right, p->right);
} 
//层次遍历
void PrintTreeByLevel(BiTree T)
{
    if(T == NULL) return;

    deque<BiTree> q;
    q.push_back(T);

    int curLevelNum; // 当前层的节点数

    while (q.size())
    {
        curLevelNum = (int)q.size(); // 此处要做类型转换，因为size()返回值为size_type类型；
        while(curLevelNum-- > 0) // 一直访问到当前层的最后一个节点
        {
            BiTree tmp = q.front();

            q.pop_front();
            printf("%d ", tmp->data);

            if(tmp->left)
                q.push_back(tmp->left);
            if(tmp->right)
                q.push_back(tmp->right);
        }
        printf("\n"); // 当前层的元素输出完毕后，换行；
    }
}
int main(){
	BiTree p, T;
	p = creat_bt();
	cout << "原树：" << endl;
	PrintTreeByLevel(p);
	
//	copyTree(T, p);
	copyTree1(T, p);
	
	cout << "新树：" << endl;
	PrintTreeByLevel(p);	
	return 0;
}

非递归

/*
复制一棵二叉树的非递归算法
*/
#include <iostream>
#include <deque>
#include <malloc.h>
using namespace std;
typedef struct BTNode {
    int data;
    struct BTNode *left, *right;
}*BiTree;

BiTree creat_bt(){ //按扩展前序建二叉树
    BiTree t;
	int x;
    cin >> x; 
    if (x==0) t=NULL;
    else 
    { 
		t = (BiTree)malloc(sizeof(BTNode));
        t->data = x;
        t->left = creat_bt();
        t->right = creat_bt();
    }
    return t;
}
void copyTree(BiTree &T, BiTree p){
	if(p == NULL)	T = NULL;
	else{
		BiTree t;
		t = T;
		deque<BiTree> q;
		q.push_back(p);
		
		int curLevelNum;
		
		while(q.size()){
			curLevelNum = (int)q.size(); // 此处要做类型转换，因为size()返回值为size_type类型；
			while(curLevelNum-- > 0) // 一直访问到当前层的最后一个节点
			{
				BiTree tmp = q.front();
				q.pop_front();
				//不用新空间
//				t = tmp;
				//用新空间
				t = (BiTree)malloc(sizeof(BTNode));
				t->data = tmp->data;

				if(tmp->left){
					//指针赋值，不用新结点
//					t->left = tmp->left;
					//创建新空间
					t->left = (BiTree)malloc(sizeof(BTNode));
					t->left->data = tmp->left->data;
				}
				if(tmp->right){
					//指针赋值，不用新结点
//					t->right = tmp->right;
					//创建新空间
					t->right = (BiTree)malloc(sizeof(BTNode));
					t->right->data = tmp->right->data;
				}
			}
		}
	}
}
//层次遍历
void PrintTreeByLevel(BiTree T)
{
    if(T == NULL) return;

    deque<BiTree> q;
    q.push_back(T);

    int curLevelNum; // 当前层的节点数

    while (q.size())
    {
        curLevelNum = (int)q.size(); // 此处要做类型转换，因为size()返回值为size_type类型；
        while(curLevelNum-- > 0) // 一直访问到当前层的最后一个节点
        {
            BiTree tmp = q.front();

            q.pop_front();
            printf("%d ", tmp->data);

            if(tmp->left)
                q.push_back(tmp->left);
            if(tmp->right)
                q.push_back(tmp->right);
        }
        printf("\n"); // 当前层的元素输出完毕后，换行；
    }
}
int main(){
	BiTree p, T;
	p = creat_bt();
	cout << "原树：" << endl;
	PrintTreeByLevel(p);
	
	copyTree(T, p);
	
	cout << "新树：" << endl;
	PrintTreeByLevel(p);	
	return 0;
}