《leetcode-php》把完美二叉树的左兄弟节点指向右兄弟节点

最新推荐文章于 2022-03-11 14:01:25 发布

原创最新推荐文章于 2022-03-11 14:01:25 发布 · 479 阅读

0 ·

CC 4.0 BY-SA版权

leetCode-php 专栏收录该内容

96 篇文章

订阅专栏

本文介绍了一种在完美二叉树中填充所有节点的next指针的方法，使其指向右兄弟节点，若无则设为NULL。通过递归算法实现，确保了常数级别的额外内存使用。

给定一个二叉树
struct TreeLinkNode {↵ TreeLinkNode *left;↵ TreeLinkNode *right;↵ TreeLinkNode *next;↵ }
填充所有节点的next指针，指向它右兄弟节点。如果没有右兄弟节点，则应该将next指针设置为NULL。
初始时，所有的next指针都为NULL
注意：

你只能使用常量级的额外内存空间
可以假设给出的二叉树是一个完美的二叉树(即，所有叶子节点都位于同一层，而且每个父节点都有两个孩子节点)。
例如：给出如下的完美二叉树 1↵ / ↵ 2 3↵ / / ↵ 4 5 6 7↵
调用完你给出的函数以后，这颗二叉树应该变成： 1 -> NULL↵ / ↵ 2 -> 3 -> NULL↵ / / ↵ 4->5->6->7 -> NULL
注：主要这个是一个完美二叉树！！！！！！

<?php
class TreeLinkNode {
    public $left  = null;
    public $right = null;
    public $next  = null;
    public $val;
    public function __construct($val) {
        $this->val = $val;
    }
}
function connect($root) {
    if ($root->left != null && $root->right != null) {
        $root->left->next = $root->right;
    }
    if ($root->next != null && $root->right != null) {
        $root->right->next = $root->next->left;
    }
    if ($root->left != null) {
        connect($root->left);
    }
    if ($root->right != null) {
        connect($root->right);
    }
    return $root;
}
$node1 = new TreeLinkNode(1);
$node2 = new TreeLinkNode(2);
$node3 = new TreeLinkNode(3);
$node4 = new TreeLinkNode(4);
$node5 = new TreeLinkNode(5);
$node6 = new TreeLinkNode(6);
$node7 = new TreeLinkNode(7);
$node1->left  = $node2;
$node1->right = $node3;
$node2->left  = $node4;
$node2->right = $node5;
$node3->left  = $node6;
$node3->right = $node7;
$root = connect($node1);
output($root);
function output($root) {
    if ($root != null) {
        print $root->val.'->'.$root->next->val.';';
    }
    if ($root->left != null) {
        output($root->left);
    }
    if ($root->right != null) {
        output($root->right);
    }
}