要将一层的结点连接起来,又需要父结点的兄弟结点的帮助,则站在第一层来连接第二层的结点。 由于第一层已经连接好,因此父结点的兄弟结点可由root->next找到 对于root的左结点(若存在):若root的右结点存在,则左结点指向右结点;若不存在,则需要root的兄弟结点uncle结点,通过uncle结点来找附近的右侧结点 对于root的右结点(若存在):直接利用uncle结点来找附近的右侧结点 以下AC代码
class Solution {
public:
Node* connect(Node* root) {
if(!root)
return NULL;
if(root->left)
root->left->next = root->right != NULL ? root->right : find(root->next);
if(root->right)
root->right->next = find(root->next);
connect(root->left);
connect(root->right);
return root;
}
Node* find(Node* uncle){
if(!uncle)
return NULL;
if(uncle->left)
return uncle->left;
if(uncle->right)
return uncle->right;
return find(uncle->next);
}
};
//Dr.He,这个代码里还有点问题,但我暂时还没有找出来,明天上完课来debug
|