829数据结构练习

1、填充每个节点的下一个右侧节点指针

c++代码：

class Solution {
public:
    Node* connect(Node* root) {
        if(!root)   return NULL;
        Node* now = new Node;
        Node* leftmost = new Node;
        now = root;
        leftmost = root;
        while(now -> left){
            now -> left -> next = now -> right; // 当前节点的左节点的next指向右节点
            if(now -> next){ //将当前节点的右节点的next指向 下一节点的左节点
                now -> right -> next = now -> next -> left;
                now = now -> next; // 更新当前节点
            }
            else{
                now = leftmost -> left; // 若当前节点没有next，更新当前节点为当前层的最左节点的左节点。
                leftmost = now;
            }
        }
        return root;
    }
};

java代码：