LeetCode116-填充每个节点的下一个右侧节点指针-完美二叉树-宽搜-数据结构

最新推荐文章于 2022-04-28 20:51:40 发布

李烦烦搞快点

最新推荐文章于 2022-04-28 20:51:40 发布

阅读量334

点赞数

分类专栏： LeetCode&&笔试题

本文链接：https://blog.csdn.net/Mr_Ghost812/article/details/123935348

版权

二叉树层次遍历节点连接算法实现数据结构

关键词由CSDN通过智能技术生成

LeetCode&&笔试题专栏收录该内容

291 篇文章 0 订阅

订阅专栏

原题链接
在这里插入图片描述

Note：

跟之前的那个按照每一层来遍历一样，不过就是记录一个最后的节点来特判一下要不要设置为空指针就好了，之后每一层的时候，让他的next指针指向队列的对头元素

代码如下：

class Solution {
public:
    Node* connect(Node* root) {
        if(!root)   return NULL;
        vector<Node*> q;
        int last = 0, f = -1, l = 0;
        q.push_back(root);

        while(f < l){
            auto p = q[++ f];
            if(p -> left)   q.push_back(p -> left), l ++;
            if(p -> right)  q.push_back(p -> right), l ++;
            if(last != f)   p -> next = q[f + 1];
            else if(last == f){
                last = l;
                p -> next = NULL;
            }
        }

        return root;
    }
};