要将一层的结点连接起来,又需要父结点的兄弟结点的帮助,则站在第一层来连接第二层的结点。
由于第一层已经连接好,因此父结点的兄弟结点可由root->next找到
对于root的左结点(若存在):若root的右结点存在,则左结点指向右结点;若不存在,则需要root的兄弟结点uncle结点,通过uncle结点来找附近的右侧结点
对于root的右结点(若存在):直接利用uncle结点来找附近的右侧结点
以下AC代码
/*
// Definition for a Node.
class Node {
public:
int val;
Node* left;
Node* right;
Node* next;
Node() : val(0), left(NULL), right(NULL), next(NULL) {}
Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}
Node(int _val, Node* _left, Node* _right, Node* _next)
: val(_val), left(_left), right(_right), next(_next) {}
};
*/
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->right);
connect(root->left);
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
connect(root->right);
connect(root->left);
先对右结点递归,再对左结点递归,是因为:先递归左结点,会一直递归到左下角,但是此时右边是空的,左边递归需要依赖右边
例如,这种情况。root->right(x)向右连接需要uncle的帮助,但uncle还没有的next还未确定,因此x的next也无法确定