题意:
给一颗完美二叉树,加一个next节点,使得next节点指向自己同层次的右边的那个节点。
思路:
当前节点的左子节点--》当前节点的右子节点
如果当前节点有兄弟节点,即:如果当前节点的next非空,那么我们当前节点的右子节点的next就是 当前节点-》next-》left。
这样通过一个简单的递归就可以解决了。
注意开始判空。
代码:
class Solution {
public:
void connect(TreeLinkNode *root) {
if(root==NULL)return;
if(root->left==NULL||root->right==NULL)return;
root->left->next = root->right;
if(root->next!=NULL){
root->right->next = root->next->left;
}
connect(root->left);
connect(root->right);
}
};