二叉排序树（插入、删除、更新、遍历、搜索、求树高。。。）

标签：二叉树   搜索   遍历   

#include <iostream>
using namespace std;
// 有序二叉树(二叉搜索树)
class Tree {
public:
	// 构造过程中初始化为空树
	Tree (void) : m_root (NULL), m_size (0) {}
	// 析构过程中销毁剩余节点
	~Tree (void) {
		clear ();
	}
	// 插入数据
	void insert (int data) {
		insert (new Node (data), m_root);
		++m_size;
	}
	// 删除第一个匹配数据，不存在返回false
	bool erase  (int data) {
		Node*& node = find (data, m_root);
		if (node) {
			// 左插右
			insert (node->m_left, node->m_right);
			Node* temp = node;
			// 右提升
			node = node->m_right;
			delete temp;
			--m_size;
			return true;
		}
		return false;
	}
	// 删除所有匹配数据
	void remove (int data) {
		while (erase (data));
	}
	// 清空树
	void clear (void) {
		clear (m_root);
		m_size = 0;
	}
	// 将所有的_old替换为_new
	void update (int _old, int _new) {
		while (erase (_old))
			insert (_new);
	}
	// 判断data是否存在
	bool find (int data) {
		return find (data, m_root) != NULL;
	}
	// 中序遍历
	void travel (void) {
		travel (m_root);
		cout << endl;
	}
	// 获取大小
	size_t size (void) {
		return m_size;
	}
	// 获取树高
	size_t height (void) {
		return height (m_root);
	}
private:
	// 节点
	class Node {
	public:
		Node (int data) : m_data (data),
			m_left (NULL), m_right (NULL) {}
		int m_data; // 数据
		Node* m_left; // 左树
		Node* m_right; // 右树
	};
	void insert (Node* node, Node*& tree) {
		if (! tree)
			tree = node;
		else if (node) {
			if (node->m_data < tree->m_data)
				insert (node, tree->m_left);
			else
				insert (node, tree->m_right);
		}
	}
	// 返回子树tree中值与data相匹配的节点的父节点中
	// 指向该节点的成员变量的别名
	Node*& find (int data, Node*& tree) {
		if (! tree)
			return tree;
		else
		if (data == tree->m_data)
			return tree;
		else
		if (data < tree->m_data)
			return find (data, tree->m_left);
		else
			return find (data, tree->m_right);
	}
	void clear (Node*& tree) {
		if (tree) {
			clear (tree->m_left);
			clear (tree->m_right);
			delete tree;
			tree = NULL;
		}
	}
	void travel (Node* tree) {
		if (tree) {
			travel (tree->m_left);
			cout << tree->m_data << ' ';
			travel (tree->m_right);
		}
	}
	size_t height (Node* tree) {
		if (tree) {
			size_t lh = height (tree->m_left);
			size_t rh = height (tree->m_right);
			return (lh > rh ? lh : rh) + 1;
		}
		return 0;
	}
	Node* m_root; // 树根
	size_t m_size; // 大小
};
int main (void) {
	Tree tree;
	tree.insert (50);
	tree.insert (70);
	tree.insert (20);
	tree.insert (60);
	tree.insert (40);
	tree.insert (30);
	tree.insert (10);
	tree.insert (90);
	tree.insert (80);
	tree.travel ();
	cout << tree.size () << ' ' << tree.height ()
		<< endl;
	tree.erase (70);
	tree.travel ();
	tree.insert (70);
	tree.insert (70);
	tree.insert (70);
	tree.travel ();
	tree.remove (70);
	tree.travel ();
	tree.insert (40);
	tree.insert (40);
	tree.travel ();
	tree.update (40, 69);
	tree.travel ();
	cout << boolalpha << tree.find (50) << endl;
	cout << tree.find (55) << endl;
	tree.clear ();
	tree.travel ();
	return 0;
}

二叉排序树（插入、删除、更新、遍历、搜索、求树高。。。）

标签：二叉树   搜索   遍历   

原文地址：http://blog.csdn.net/liyuan_669/article/details/39255851