Leetcode 二叉树填充每个节点的下一个右侧节点指针

这种题给出了两种情况,第一种是完美二叉树(特殊情况),第二种是一般情况下的二叉树。
二叉树填充每个节点的下一个右侧节点指针
对于特殊情况会有一些特殊的解法。我仅就我尝试过的几种方法做一个记录。

首先是对于这种题目的通解(对于两种情况都适用):层序遍历记录每层节点,最后再倒序连接起来即可。

/*
// 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)

当然还有一些其它比较巧妙的解法,后面自己会再对比一下。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值