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

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

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

1. 题目

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

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

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

类似题目:LeetCode 116

2. 解题

2.1 递归

左节点:

  • root有左节点和右节点,则左节点的next为右节点
  • root右节点为null,则查找父节点的兄弟节点的最左边子元素

右节点:

  • root右节点不为null,其next为父节点的兄弟节点的最左边子元素

递归:要先构建右子树,再构建左子树,因为寻找父节点的兄弟节点是从左到右遍历的,如果右子树next没接上就遍历,会出错

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

2.2 queue循环

以下代码也可以解LeetCode 116

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

2.3 利用next循环

代码语言:javascript
复制
class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL)
        	return root;
        Node *parent = root, *prev, *tmp;
        while(parent)
        {
        	while(parent && !parent->left && !parent->right)
        		parent = parent->next;//找到第一个有子的节点parent
        	if(parent == NULL)
        		break;
        	prev = NULL;
        	tmp = parent;
        	while(tmp)	//遍历parent层,将其下层连接
        	{
        		if(tmp->left)
        		{
        			if(prev)
        				prev->next = tmp->left;
        			prev = tmp->left;
        		}
        		if(tmp->right)
        		{
        			if(prev)
        				prev->next = tmp->right;
        			prev = tmp->right;
        		}	
        		tmp = tmp->next;
        	}
        	parent = parent->left ? parent->left : parent->right;
        	//parent移向下一层
        }
        return root;
    }
};
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019/10/11 ,如有侵权请联系 cloudcommunity@tencent.com 删除

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

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

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

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