欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode 题目-671. 二叉树中第二小的节点/687.最长同值路径(python实现)

程序员文章站 2024-02-18 09:06:52
...

作为要准备踏入码农行业的人来说,要准备校招,怎么能不去刷刷LeetCode呢?

671. 二叉树中第二小的节点

  • 题目要求

给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话,那么这个节点的值不大于它的子节点的值。

给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -1 。

  • 例如
输入: 
    2
   / \
  2   5
     / \
    5   7

输出: 5
说明: 最小的值是 2 ,第二小的值是 5 。


输入: 
    2
   / \
  2   2

输出: -1
说明: 最小的值是 2, 但是不存在第二小的值。
  • 分析

方法:层次遍历将不等于根节点的值都放入一个列表中,结果返回列表的最小值即可。(题目:所有子节点值都大于等于根节点)

class Solution(object):
    def findSecondMinimumValue(self, root):
        val=root.val
        nodes=[]
        nodes.append(root)
        nums=[]
        while nodes:
            length =len(nodes)
            for i in range(length):
                node =nodes.pop()
                if node.val!=val:
                    nums.append(node.val)
                if node.left:
                    nodes.append(node.left)
                    nodes.append(node.right)
        if nums:
            return min(nums)
        return -1

作为要准备踏入码农行业的人来说,要准备校招,怎么能不去刷刷LeetCode呢?

687.最长同值路径

  • 题目要求

给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。

注意:两个节点之间的路径长度由它们之间的边数表示。

  • 例如
     5
             / \
            4   5
           / \   \
          1   1   5
输出:

2

输入:

              1
             / \
            4   5
           / \   \
          4   4   5
输出:

2
  • 分析

方法:深度遍历,传递当前节点的值,判断当前节点的左右子树的最长路径,并用一值记录左右子树做长路径之和。


class Solution(object):
    def longestUnivaluePath(self, root):
        if not root:
            return 0
        self.max_num=0
        self.search(root,root.val)
        return self.max_num
    def search(self,root,val):
        if root is None:
            return 0
        left=self.search(root.left,root.val)
        right=self.search(root.right,root.val)
        self.max_num=max(self.max_num,left+right)
        if root.val==val:
            return max(left,right)+1
        return 0