很自然的推广,假设去掉全然二叉树的条件呢?由于这个条件不是关键,因此不会影响整体的思路。做法依旧是每次找到一层的起点,然后一层一层的走。
假设是全然二叉树的话,每层的起点就是上一层起点的左孩子,兄弟之间的链接也简单,直接找相应父亲的左右孩子就可以。
一般二叉树时,起点应该是上一层第一个有孩子节点的左孩子,或者没有左孩子时。是他的右孩子。
为了能在孩子层中不断链接,我们必须保存当孩子层的前一个节点,当当前层找到一个节点有孩子,就接到这个pre节点后面,然后更新pre节点的指向。
class Solution { public: void connect(TreeLinkNode *root) { if(root == NULL) return; TreeLinkNode *beginNode = root, *leftNode, *pre; while(beginNode){ leftNode = beginNode; pre = NULL; beginNode = NULL; while(leftNode){ if(leftNode->left&&leftNode->right){ leftNode->left->next = leftNode->right; if(pre) pre->next = leftNode->left; if(!beginNode) beginNode = leftNode->left; pre = leftNode->right; }else if(leftNode->left){ if(pre) pre->next = leftNode->left; if(!beginNode) beginNode = leftNode->left; pre = leftNode->left; }else if(leftNode->right){ if(pre) pre->next = leftNode->right; if(!beginNode) beginNode = leftNode->right; pre = leftNode->right; } leftNode = leftNode->next; } } } };