No.130 - LeetCode117 - 层次链

ShellDawn

于 2019-10-10 10:25:11 发布

阅读量204

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： LeetCode

本文链接：https://blog.youkuaiyun.com/ShellDawn/article/details/102476011

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用队列实现的二叉树节点连接算法，该算法通过层级遍历的方式，将同一层级的相邻节点的next指针进行连接，从而实现节点之间的横向连接。此算法适用于具有next指针的二叉树结构。

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

/*
// Definition for a Node.
class Node {
public:
    int val;
    Node* left;
    Node* right;
    Node* next;

    Node() {}

    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;
        queue<Node*> q[2];
        int A = 0; // 当前层
        int B = 1; // 下一层
        q[A].push(root);
        while(!q[A].empty()){
            while(!q[A].empty()){
                Node* t = q[A].front();
                q[A].pop();
                if(!q[A].empty()) t->next = q[A].front();
                if(t->left != NULL) q[B].push(t->left);
                if(t->right != NULL) q[B].push(t->right);
            }
            swap(A,B);
        }
        return root;
    }
};