DFS&遍历 Populating Next Right Pointers in Each Node

最新推荐文章于 2023-03-11 16:53:38 发布

原创最新推荐文章于 2023-03-11 16:53:38 发布 · 540 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

234 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）的算法，用于解决树结构中节点之间的连接问题。该算法的时间复杂度为O(N)，空间复杂度优化至O(lgN)，特别适用于树形结构数据的遍历与节点连接操作。

思路：

方法一：DFS。

时间复杂度O(N)，空间复杂度O(lgN)。

由于递归需要栈，所以空间复杂度没有达到题目要求的O(1)。

/**
 * Definition for binary tree with next pointer.
 * struct TreeLinkNode {
 *  int val;
 *  TreeLinkNode *left, *right, *next;
 *  TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
 * };
 */
class Solution {
public:
    void connect(TreeLinkNode *root) {
        if(root == NULL || root->left == NULL) return;
        TreeLinkNode *cur;
        while(root->left != NULL) {
            cur = root;
            while(cur != NULL) {
                cur->left->next = cur->right;
                if(cur->next != NULL) {
                    cur->right->next = cur->next->left;
                }
                cur = cur->next;
            }
            root = root->left;
        }
    }
};

注意如果是单root节点，直接返回，因为：

Initially, all next pointers are set to NULL.

否则的话层层遍历，因为上一层的next设置好了，下一层就可以直接通过上一层的next到sibling节点上，且第一层的next可以直接设置，因为就有left和right两个节点。