C++线索化二叉树

#include <iostream>
using namespace std;

enum PointerTag {THREAD, LINK};

template<class T>
struct BinaryTreeNodeThd
{
	T _data;                      //数据
	BinaryTreeNodeThd<T>* _left;  //左孩子
	BinaryTreeNodeThd<T>* _right; //右孩子
	PointerTag _leftTag;          //左孩子线索标志  
	PointerTag _rightTag;         //右孩子线索标志  

	BinaryTreeNodeThd(const T& data)
		:_data(data)
		,_left(NULL)
		,_right(NULL)
		,_leftTag(LINK)
		,_rightTag(LINK)
	{}
};


template<class T>
class BinaryTreeThd
{
public:
	BinaryTreeThd(const T* array, size_t size, const T& invalid)
	{
		size_t index = 0;
		_root = _CreateTree(array, size, index, invalid);
	}

	~BinaryTreeThd()
	{
		_DestroyTree(_root);
		_root = NULL;
	}

	void InOrderThreading()
	{
		BinaryTreeNodeThd<T>* prev = NULL;

		_InOrderThreading(_root, prev);
	}

	void PreOrderThreading()
	{
		BinaryTreeNodeThd<T>* prev = NULL;

		_PreOrderThreading(_root, prev);
	}

	void PostOrderThreading()
	{
		BinaryTreeNodeThd<T>* prev = NULL;

		_PostOrderThreading(_root, prev);
	}

	void PreOrderThd()
	{
		BinaryTreeNodeThd<T>* cur = _root;

		while (cur)
		{
			while (cur && LINK == cur->_leftTag)
			{
				cout<<cur->_data<<" ";
				cur = cur->_left;
			}

			cout<<cur->_data<<" ";

			cur = cur->_right;
		}

		cout<<endl;
	}

	void InOrderThd()
	{
		BinaryTreeNodeThd<T>* cur = _root;

		while (cur)
		{
			while (cur && LINK == cur->_leftTag)
			{
				cur = cur->_left;
			}

			cout<<cur->_data<<" ";

			while (THREAD == cur->_rightTag)
			{
				cur = cur->_right;
				cout<<cur->_data<<" ";
			}

			cur = cur->_right;
		}

		cout<<endl;
	}

protected:
	BinaryTreeNodeThd<T>* _CreateTree(const T* array, size_t size, size_t& index, const T& invalid)
	{
		BinaryTreeNodeThd<T>* root = NULL;
		if (index < size && array[index] != invalid)
		{
			root = new BinaryTreeNodeThd<T>(array[index]);
			root->_left = _CreateTree(array, size, ++index, invalid);
			root->_right = _CreateTree(array, size, ++index, invalid);
		}

		return root;
	}

	void _DestroyTree(BinaryTreeNodeThd<T>* root)
	{
		if (NULL == root)
			return;

		if (LINK == root->_leftTag)
			_DestroyTree(root->_left);

		if (LINK == root->_rightTag)
			_DestroyTree(root->_right);

		delete root;
	}
	
	void _PreOrderThreading(BinaryTreeNodeThd<T>* cur, BinaryTreeNodeThd<T>*& prev)
	{
		if (NULL == cur)
			return;

		if (NULL == cur->_left)
		{
			cur->_leftTag = THREAD;
			cur->_left = prev;
		}

		if (prev && NULL == prev->_right)
		{
			prev->_rightTag = THREAD;
			prev->_right = cur;
		}

		prev = cur;

		if (cur->_leftTag == LINK)
			_PreOrderThreading(cur->_left, prev);

		if (cur->_rightTag == LINK)
			_PreOrderThreading(cur->_right, prev);
	}

	void _InOrderThreading(BinaryTreeNodeThd<T>* cur, BinaryTreeNodeThd<T>*& prev)
	{
		if (NULL == cur)
			return;

		_InOrderThreading(cur->_left, prev);

		if (NULL == cur->_left)
		{
			cur->_leftTag = THREAD;
			cur->_left = prev;
		}

		if (prev && NULL == prev->_right)
		{
			prev->_rightTag = THREAD;
			prev->_right = cur;
		}

		prev = cur;

		_InOrderThreading(cur->_right, prev);
	}

	void _PostOrderThreading(BinaryTreeNodeThd<T>* cur, BinaryTreeNodeThd<T>*& prev)
	{
		if (NULL == cur)
			return;

		_PostOrderThreading(cur->_left, prev);
		_PostOrderThreading(cur->_right, prev);

		if (cur->_left == NULL)
		{
			cur->_leftTag = THREAD;
			cur->_left = prev;
		}

		if (prev && NULL == prev->_right)
		{
			prev->_rightTag = THREAD;
			prev->_right = cur;
		}

		prev = cur;
	}
protected:
	BinaryTreeNodeThd<T>* _root;
};


void Test()
{
	int a[] = {1, 2, 3, '#', '#', 4, '#', '#', 5, 6};
	BinaryTreeThd<int> t1(a, sizeof(a)/sizeof(a[0]), '#');

	t1.PreOrderThreading();
	t1.PreOrderThd();

	BinaryTreeThd<int> t2(a, sizeof(a)/sizeof(a[0]), '#');
	t2.InOrderThreading();
	t2.InOrderThd();

	int a1[] = {1, 2, '#', 3, '#', '#', 4, 5, '#', 6, '#', 7, '#', '#', 8};
	BinaryTreeThd<int> t3(a1, sizeof(a1)/sizeof(a1[0]), '#');
	t3.PreOrderThreading();
	t3.PreOrderThd();
}

int main()
{
	Test();
	return 0;
}