Populating Next Right Pointers in Each Node I II@LeetCode
Populating Next Right Pointers in Each Node I
树的广度优先搜索题。记录下每一层的节点总个数,然后根据广度优先搜索的原则进行遍历,将非null
节点都加入到队列中,对于同一层中的节点,将其next
指向队列中的下一个节点即可。
实现代码:
java
public class Solution { public void connect(TreeLinkNode root) { if (root == null) return; LinkedList<TreeLinkNode> nodes = new LinkedList<TreeLinkNode>(); nodes.add(root); int numOfLevelTotal = 1; while (!nodes.isEmpty()) { TreeLinkNode treeLinkNode = nodes.poll(); numOfLevelTotal--; if (treeLinkNode.left != null) { nodes.add(treeLinkNode.left); } if (treeLinkNode.right != null) { nodes.add(treeLinkNode.right); } if (numOfLevelTotal > 0) { treeLinkNode.next = nodes.getFirst(); } else { numOfLevelTotal = nodes.size(); } } } }
Populating Next Right Pointers in Each Node II
根据题目来说,这一题和上一次的区别在于:
What if the given tree could be any binary tree? Would your previous solution still work?
但是由于之前所采用的方法并没有这种局限,所以直接拷贝过来也可以AC
。
不过这里存在一个问题,仔细看题目里的要求:
You may only use constant extra space.
理论上,采用队列的话是肯定没办法只是用常数额外内存的,但是LeetCode好像在这件事上没有检测的这么严,起码我写的Java代码和我看到的用递归解决的C++代码都可以通过。
那么如果硬要纠结一下这一条呢?不使用队列怎么做?
其实也不难,只是思路要转变一下,就不能是遍历这一层同时连接这一层,而是遍历这一层连接下一层。那么比较重要的就是要记录每一层的头节点,由于每一层在被遍历的时候是已经连接好了的,所以不必担心找不到节点的问题,如果挨个找寻next
节点的子节点即可,子节点先内部(相同父节点)连接,然后再连接到总链表中即可。
实现代码:
java
public class Solution { public void connect(TreeLinkNode root) { TreeLinkNode levelHead = root, nextLevelHead = null; while (levelHead != null) { TreeLinkNode node = levelHead, tail = null; while (node != null) { if (node.left != null && node.right != null) { node.left.next = node.right; } TreeLinkNode sub; if (node.left != null) sub = node.left; else if (node.right != null) sub = node.right; else sub = null; if (sub != null) { if (nextLevelHead == null) { nextLevelHead = sub; tail = sub; } else { tail.next = sub; } while (tail.next != null) tail = tail.next; } node = node.next; } levelHead = nextLevelHead; nextLevelHead = null; } } }
1.1k 声望
63 粉丝
推荐阅读
关于 C++ vector 的两个小 tips
本来这篇文章标题我想起成《关于 vector 的两个小坑》,后来想想,其实也不算是坑,还是自己对原理性的东西理解的没做那么透彻。工作中遇到的很多问题,后来归根到底都是基础不牢靠。
findingea阅读 1.4k
刷完15道js版dp题,面试再也不怕了
某个男人 动态规划,而我作为一个致力称为厨师界最会写算法的前端,总得刷上一部分题,有那么一点发现吧,现在我们就来聊聊,菜鸡如我,发现了什么。
hellocoder2028赞 1阅读 360
力扣之 4 的幂 & 3 的幂 & 2 的幂(递归思想)
题目描述给定一个整数,写一个函数来判断它是否是 4 的幂次方。如果是,返回 true ;否则,返回 false 。整数 n 是 4 的幂次方需满足:存在整数 x 使得 n == 4x示例 1: {代码...} 示例 2: {代码...} 示例 3: {...
水冗水孚赞 1阅读 678
力扣之仅仅反转字母(双指针方式)
题目描述给你一个字符串 s ,根据下述规则反转字符串:所有非英文字母保留在原有位置。所有英文字母(小写或大写)位置反转。返回反转后的 s 。示例 1: {代码...} 示例 2: {代码...} 示例 3: {代码...} 力扣...
水冗水孚赞 1阅读 528
力扣之回文数(双指针中的对撞指针公式模板)
什么双指针没刷算法之前,一听双指针,感觉很厉害的样子。实际上呢?也的确是一个不错的解题思路方式。在LeetCode上的双指针是一大类题目的解决方式,看一下,发现有近200题是双指针类型的,如下图:由此可见,双...
水冗水孚阅读 965
力扣之反转字符串之原地修改输入数组(双指针方式)
题目描述编写一个函数,其作用是将输入的字符串反转过来。输入字符串以字符数组 s 的形式给出。不要给另外的数组分配额外的空间,你必须原地修改输入数组、使用 O(1) 的额外空间解决这一问题。示例 1: {代码...}...
水冗水孚阅读 881
代码随想录打卡day2
977,有序数组的平方排序 {代码...} 209,长度最小的子序列方法:滑动窗口 {代码...} 59,螺旋矩阵方法:找规律 {代码...}
赵钱什么李阅读 735
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。