继续刷题,题目如下图
如果用C描述的话,就是一个二叉树节点定义包括右节点指针,左节点指针,和右相连指针;给出一个二叉树,维护其右相邻指针,如果是最右边节点,则指针为空。
struct Node {
int val;
Node *left;
Node *right;
Node *next;
} 
思路其实很简单,这个可以按层分析二叉树,首先把当前层节点按照从左到右放入一个队列中,遍历这个队列;如果不是队列最后一个节点,则按照当前节点的next就是下一个节点;同时把每一个节点的子节点按照从左到右放入下一层队列;然后 遍历下一层队列;直到这个队列为空,遍历完成。
代码如下:
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
"""# Definition for a Node.class Node: def __init__(self, val: int = 0, left: ‘Node‘ = None, right: ‘Node‘ = None, next: ‘Node‘ = None): self.val = val self.left = left self.right = right self.next = next"""class Solution: def connect(self, root: ‘Node‘) -> ‘Node‘: if root == None: return None else: nodeList = [] nodeList.append(root) while nodeList != []: nextList = [] for i in range(len(nodeList)): if nodeList[i].left != None: nextList.append(nodeList[i].left) if nodeList[i].right != None: nextList.append(nodeList[i].right) if i!= len(nodeList)-1: nodeList[i].next = nodeList[i+1] nodeList = nextList return root
|
原文:https://www.cnblogs.com/chenguopa/p/15239896.html