Populating Next Right Pointers in Each Node I II@LeetCode

Populating Next Right Pointers in Each Node I

树的广度优先搜索题。记录下每一层的节点总个数,然后根据广度优先搜索的原则进行遍历,将非null节点都加入到队列中,对于同一层中的节点,将其next指向队列中的下一个节点即可。

实现代码:

javapublic 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节点的子节点即可,子节点先内部(相同父节点)连接,然后再连接到总链表中即可。

实现代码:

javapublic 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 粉丝
0 条评论
推荐阅读
关于 C++ vector 的两个小 tips
本来这篇文章标题我想起成《关于 vector 的两个小坑》,后来想想,其实也不算是坑,还是自己对原理性的东西理解的没做那么透彻。工作中遇到的很多问题,后来归根到底都是基础不牢靠。

findingea阅读 1.4k

刷完15道js版dp题,面试再也不怕了
某个男人 动态规划,而我作为一个致力称为厨师界最会写算法的前端,总得刷上一部分题,有那么一点发现吧,现在我们就来聊聊,菜鸡如我,发现了什么。

hellocoder20281阅读 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

1.1k 声望
63 粉丝
宣传栏