二叉树的重建


上面是编程之美上的解法,下面是自己的解法:

#include <iostream>
#include <queue>

using namespace std;

struct Node 
{
	Node(int i = -1, Node *pLeft = NULL, Node *pRight = NULL) : data(i), left(pLeft), right(pRight) {}
	int data;
	Node *left;
	Node *right;
};

const char *preOrder = "12473568";
const char *inOrder = "47215386";

void printLevel(Node *root)
{
	if (root == NULL)
		return;

	queue<Node *> nqueue;
	nqueue.push(root);

	int start = 0;
	int last = 1;
	int tempLast = last;
	while (!nqueue.empty())
	{
		Node *top = nqueue.front();
		nqueue.pop();
		cout << top->data << " ";

		if (top->left)
		{
			nqueue.push(top->left);
			last++;
		}
		if (top->right)
		{
			nqueue.push(top->right);
			last++;
		}

		start++;
		if (start == tempLast)
		{
			tempLast = last;
			cout << endl;
		}
	}
}

Node *_construct(const char *preOrder, const int preLength,
				 const char *inOrder, const int inLength)
{
	if (preOrder == NULL || preLength <= 0 || inOrder == NULL || inLength <= 0)
		return NULL;

	int pos = inLength;
	for (int i = 0; i < inLength; i++)
	{
		if (inOrder[i] == *preOrder)
		{
			pos = i;
			break;
		}
	}
	if (pos == inLength)
	{
		cout << "wrong" << endl;
		return NULL;
	}
	cout << *preOrder << " ";
	Node *newNode = new Node(*preOrder - '0');
	newNode->left = _construct(preOrder + 1, preLength - 1, inOrder, pos);
	newNode->right = _construct(preOrder + pos + 1, preLength - 1, inOrder + pos + 1, inLength - pos - 1);

	return newNode;
}

Node *construct(const char *preOrder, const char *inOrder)
{
	if (preOrder == NULL || inOrder == NULL)
		return NULL;

	int preLength = strlen(preOrder);
	int inLength = strlen(inOrder);

	return _construct(preOrder, preLength, inOrder, inLength);
}

void preTranverse(Node *root)
{
	if (root == NULL)
		return;
	cout << root->data << " ";
	preTranverse(root->left);
	preTranverse(root->right);
}

void main()
{
	Node *root = construct(preOrder, inOrder);
	cout << endl;

	if (root != NULL)
	{
		preTranverse(root);
	}
	else
	{
		cout << "wrong" << endl;
	}
}


评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值