思路:
解法一:迭代+层序遍历
本质相同,也是考察二叉树层序遍历算法,这里用DFS实现。只不过多一步填充next。
遍历本层结点时,若当前结点右侧还有结点,也就是队列非空,且当前结点不是本层最后一个结点(这个时候之前若有结点有孩子,孩子已入队,队列非空,但不应填充next指针) ,则填充next指针。
/*
// 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) {
queue<Node*>q;
if(root){q.push(root);}//根结点非空则入队
while(!q.empty()){//队列非空时
int size=q.size();
for(int i=0;i<size;i++){//遍历本层
Node*cur=q.front();取当前结点
q.pop();
if(!q.empty()&&i!=size-1){cur->next=q.front();}//当前结点右侧还有结点,填充next
if(cur->left){q.push(cur->left);}//左右孩子非空则入队
if(cur->right){q.push(cur->right);}
}
}
return root; //原地改变,直接返回根结点即可
}
};
解法二:使用已建立的 next指针
可以充分利用完美二叉树性质,所有叶子节点都在同一层,每个父结点都有两个孩子。
从根结点开始,用leftmost记录当前层最左结点,当前指针head作为父结点,逐层遍历为下一层的结点填充next指针。
我们一共要填充两种next指针,一种是同一个父结点的两个结点之间,一种是不同父结点的两个结点之间。第一种可直接建立,让head的左孩子的next指向右孩子;第二种需判断head是否有next,若有,则让head的右孩子指向head的next的左孩子。
head不断在本层后移,当head走到空时说明本层已遍历完,可以开始下一层,leftmost不断跳到下一层最左结点,直到走到空说明所有层都已经填充完毕。
class Solution {
public:
Node* connect(Node* root) {
if(!root) return nullptr;
Node*leftmost=root;//从根结点开始
while(leftmost->left){//当左结点不空说明还有可连接的结点
Node*head=leftmost;// 遍历这一层结点,为下一层的节点更新 next 指针
while(head){
head->left->next=head->right;//连接同一个父节点的两个子节点
if(head->next){//在不同父亲的子节点之间建立连接
head->right->next=head->next->left;
}
head=head->next;//指针后移
}
leftmost=leftmost->left;//去下一层最左结点
}
return root;
}
};
思路:
本题与上一题的区别就是本题中给的二叉树不是完美二叉树,所以上一题的解法二不能用,用解法一同样可以解决。
代码注释同上题。