面试经典 150 题 7 —(二叉树)— 117. 填充每个节点的下一个右侧节点指针 II

本文介绍了一种使用广度优先搜索(BFS)算法,通过队列操作,为二叉树的每个节点填充下一个右侧节点指针的方法,实现前驱节点的连接。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

117. 填充每个节点的下一个右侧节点指针 II

在这里插入图片描述

方法:广度优先搜索BFS

涉及二叉树的搜索可能用到的:前中后序遍历(用栈)、深度优先搜索DFS(用栈)、广度优先搜索BFS(用队列)

/*
// 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 == NULL){
            return NULL;
        }
        // 广度优先搜索利用队列,可以直接用STL的queue
        queue<Node*> nodeQueue;
        nodeQueue.push(root);
        while(nodeQueue.size()){
            // 每一层的数量
            int levelCount = nodeQueue.size();
            // 前一个节点;从右往左依次向右指
            Node* pre = NULL;
            for(int i = 0; i < levelCount; i++){
                // 每次取出队首的元素,将next指向其前一个节点
                Node* queueHead = nodeQueue.front();
                nodeQueue.pop();
                queueHead->next = pre;
                pre = queueHead;
                // 判断它的右左孩子;再把他的右孩子、左孩子加入到队列中,从右往左依次向右指
                if(queueHead->right){
                    nodeQueue.push(queueHead->right);
                }
                if(queueHead->left){
                    nodeQueue.push (queueHead->left);
                }
            }
        }
        return root;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值