比如:
0
1 2
3 4 5
6 7
1->2
3->4->5
6->7
参考数据结构:
struct TREE_NODE
{
TREE_NODE(int n) : nVal(n), pLft(NULL), pRgt(NULL), pSib(NULL)
{}
int nVal;
TREE_NODE* pLft;
TREE_NODE* pRgt;
TREE_NODE* pSib;
0
1 2
3 4 5
6 7
1->2
3->4->5
6->7
参考数据结构:
struct TREE_NODE
{
TREE_NODE(int n) : nVal(n), pLft(NULL), pRgt(NULL), pSib(NULL)
{}
int nVal;
TREE_NODE* pLft;
TREE_NODE* pRgt;
TREE_NODE* pSib;
};
在想这个题目的时候,有两个难点无法克服:
1. 左子树在递归过程中,需要用到右子树,所以要求右子树先递归遍历,最后在递归左子树
2. 如果一个节点没有子节点,如果将前一个节点的指针指向他后一个指针。
下面的算法可以将上面的两个问题解决
struct Node {
Node *left;
Node *right;
Node *sib;
};
Node *find_next(Node *root) {
while (root = root->sib) {
if (root->left | root->right)
return root->left ? root->left : root->right;
}
return NULL;
}
void linkLevelNodes(Node *root) {
if (root == NULL)
return;
if (root->left)
root->left->sib = root->right ? root->right : find_next(root);
if (root->right)
root->right->sib = find_next(root);
linkLevelNodes(root->right);
linkLevelNodes(root->left);
}