Leetcode 116. 填充每个节点的下一个右侧节点指针解题思路及C++实现

最新推荐文章于 2021-10-19 20:45:54 发布

原创

最新推荐文章于 2021-10-19 20:45:54 发布 · 499 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了两种方法解决LeetCode 116问题，即如何填充每个节点的下一个右侧节点指针。方法一是通过层序遍历来更新指针，利用两个栈来简化操作。方法二是利用完美二叉树的特性，通过递归或迭代方式高效设置左右子节点的next指针。递归方法虽然简洁，但消耗资源较多。

方法一：层序遍历（这是比较暴力愚蠢的方法）

解题思路：

这里使用层序遍历访问这颗完美二叉树，使用的是两个栈，而不是两个队列，因为这样在遍历每一层并指定next指针时会更方便一些，但是要注意的一点是，循环内的临时栈s2 是先push右子节点，再push左子节点。s2中节点顺序是反过来的，所以要依次pop到s1中。

/*
// 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 root;
        //层序遍历
        stack<Node*> s1;
        s1.push(root);
        while(!s1.empty()){
            stack<Node*> s2;
            Node* temp1 = NULL;
            Node* temp2 = NULL;
            while(!s1.empty()){
                temp1 = s1.top();
                s1.pop();
                temp1->next = temp2;
                temp2