C之一个二叉树题目解答

那天看到这么一个题目，内容是这样的：
[quote]编写一算法，利用叶子结点中的空指针域将所有叶子结点链接为一个双链表，算法返回头结点的地址。[/quote]

我大概思考了一下，首先得先得到这棵树的一个遍历，然后树中节点的左子树指针相当于双向链表中的prior指针，右子树指针相当于双向链表的next指针，那么头结点应该就是叶子中最左的，我们前面得到的遍历结果中，凡是左右子树都为NULL的就是叶子，只需要将这些链接到双向链表中就OK了。

我给出的伪码如下：

/*
 * 假设二叉树的结点定义 如下：
 * struct TNode {
 *     TNode* rChild;
 *     TNode* lChild;
 *     T value;
 * };
 */
TNode* linkLeaf(TNode* pRoot) {

	Stack auxiliaryStack;
	Stack inorderStack;
	TNode* p=pRoot;

	while (true) {
		if (p) {
			auxiliaryStack.push(p);
			p = p->lChild;
		} else {
			if (auxiliaryStack.size()) {
				p = auxiliaryStack.pop();
				inorderStack.push(p);
				p = p->rChild;
			} else {
				break;
			}

		}
	}

	TNode* pCurLeaf = NULL;
	while (inorderStack.size()) {
		p = inorderStack.pop();
		if (!p->rChild&&!p->lChild) {
			if (!pCurLeaf) {
				pCurLeaf = p;
			} else {
				p->rChild = pCurLeaf;
				pCurLeaf->lChild = p;
				pCurLeaf = p;
			}
		}

	}

    return pCurLeaf;

}