前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 116. 填充每个节点的下一个右侧节点指针(递归&循环)

LeetCode 116. 填充每个节点的下一个右侧节点指针(递归&循环)

作者头像
Michael阿明
发布2020-07-13 15:28:37
4530
发布2020-07-13 15:28:37
举报

1. 题目

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

代码语言:javascript
复制
struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。

初始状态下,所有 next 指针都被设置为 NULL

在这里插入图片描述
在这里插入图片描述

2. 解题

2.1 递归

代码语言:javascript
复制
class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL || root->left == NULL)
        	return root;
        root->left->next = root->right;
    	if(root->next != NULL)
    		root->right->next = root->next->left;
    	connect(root->left);
    	connect(root->right);
        return root;
    }
};

2.2 循环

代码语言:javascript
复制
class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL || root->left == NULL)
        	return root;
        queue<Node*> q;
        q.push(root);
        Node *p;
        int n;
        while(!q.empty())
        {
        	n = q.size();
        	while(n--)
        	{
        		p = q.front();
	        	if(p->left != NULL)
	        	{
	        		q.push(p->left);
	        		q.push(p->right);
		        	p->left->next = p->right;
			    	if(p->next != NULL)
			    		p->right->next = p->next->left;
			    }
	        	q.pop();
        	}	
        }
        return root;
    }
};

2.3 O(1)空间复杂度

  • 利用next指针
代码语言:javascript
复制
class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL || root->left == NULL)
        	return root;
        Node *parent = root, *lvfirst, *prev;
        while(parent != NULL)
        {
        	lvfirst = parent->left;
        	if(lvfirst != NULL)
        	{
        		while(parent)
	        	{
	        		parent->left->next = parent->right;
	        		prev = parent->right;
	        		parent = parent->next;
	        		if(parent)
	        			prev->next = parent->left;
	        	}
        	}
        	parent = lvfirst;	
        }
        return root;
    }
};
在这里插入图片描述
在这里插入图片描述
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019/10/11 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
    • 2.1 递归
      • 2.2 循环
        • 2.3 O(1)空间复杂度
        领券
        问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档