填充二叉树节点的右侧指针(c++&&golang)

最新推荐文章于 2024-10-31 18:26:46 发布

attempt_to_do

最新推荐文章于 2024-10-31 18:26:46 发布

阅读量480

点赞数

分类专栏：数据结构和算法文章标签： c++ golang 算法

本文链接：https://blog.csdn.net/weixin_43021844/article/details/122194153

版权

数据结构和算法专栏收录该内容

15 篇文章 0 订阅

订阅专栏

/*
给定一个 完美二叉树 ，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：

struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。

初始状态下，所有 next 指针都被设置为 NULL。
*/

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

    Node() : val(0), left(NULL), right(NULL), next(NULL) {}

    Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}

    Node(int _val, Node *_left, Node *_right, Node *_next)
        : val(_val), left(_left), right(_right), next(_next) {}
};

#include <bits/stdc++.h>
using namespace std;

class Solution
{
public:
    void connectWoNode(Node *node1, Node *node2)
    {
        if (node1 == nullptr || node2 == nullptr)
        {
            return;
        }
        node1->next = node2;
        connectWoNode(node1->left, node1->right);
        connectWoNode(node2->left, node2->right);
        connectWoNode(node1->right, node2->left);
    }
    Node *connect(Node *root)
    {
        if (root == nullptr)
        {
            return nullptr;
        }
        root->next = nullptr;
        connectWoNode(root->left, root->right);
        return root;
    }
};

package main

type Node struct {
	Val   int
	Left  *Node
	Right *Node
	Next  *Node
}

func connect(root *Node) *Node {
	if root == nil {
		return nil
	}
	var dfs func(*Node, *Node)
	dfs = func(node1 *Node, node2 *Node) {
		//结束条件
		if node1 == nil || node2 == nil {
			return
		}

		node1.Next = node2
		//连接左子树的左右节点
		dfs(node1.Left, node1.Right)
		//连接右子树的左右节点
		dfs(node2.Left, node2.Right)
		//连接左子树的右节点和右子树的左节点
		dfs(node1.Right, node2.Left)
	}
	dfs(root.Left, root.Right)
	return root
}