这种题给出了两种情况,第一种是完美二叉树(特殊情况),第二种是一般情况下的二叉树。
二叉树填充每个节点的下一个右侧节点指针
对于特殊情况会有一些特殊的解法。我仅就我尝试过的几种方法做一个记录。
首先是对于这种题目的通解(对于两种情况都适用):层序遍历记录每层节点,最后再倒序连接起来即可。
/*
// 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;
queue<Node*> que;
vector<Node*> vec;
que.push(root);
while(!que.empty())
{
int len = que.size();
while(len--)
{
Node* node = que.front();
que.pop();
if(node->left) que.push(node->left);
if(node->right) que.push(node->right);
vec.push_back(node);
}
for(int i = vec.size()-1;i>0;i--)
{
vec[i-1]->next = vec[i];
}
vec.clear();//每层记录完后清空vector
}
return root;
}
};
对于完美二叉树的情况还有下面几种解法:(参考了这两题题解里面的一些大佬的思路)
1、思路:
每个node的左子树的next都与node的右子树相连;
每个node的右子树的next都与node的next的左子树相连。
这样思考就可以得到下面的代码:
class Solution {
public:
Node* connect(Node* root) {
search(root, NULL);
return root;
}
void search(Node* node, Node* next)
{
if(node!=NULL)
{
node->next = next;
search(node->left,node->right);
search(node->right, node->next != NULL ? node->next->left : NULL);
}
}
};
2、拉拉链法
以root为中心连接左右子树,然后递归即可。
代码如下:
class Solution {
public:
Node* connect(Node* root) {
if(!root || !root -> left) return root;
root -> left -> next = root -> right;
Node* now = new Node;
now = root -> left;
while(now -> left){
now -> right -> next = now -> next -> left;
now = now -> right;
}
root -> left = connect(root -> left);
root -> right = connect(root -> right);
return root;
}
};
作者:youlookdeliciousc
链接:https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node/solution/c-die-dai-di-gui-shuang-jie-la-lian-fa-you-tu-you-/
来源:力扣(LeetCode)
当然还有一些其它比较巧妙的解法,后面自己会再对比一下。