Python 解LeetCode：671. Second Minimum Node In a Binary Tree

题目在这里，要求一个二叉树的倒数第二个小的值。二叉树的特点是父节点的值会小于子节点的值，父节点要么没有子节点，要不左右孩子节点都有。

分析一下，根据定义，跟节点的值肯定是二叉树中最小的值，剩下的只需要找到左右子树中比跟节点大的最小值就可以了。对于这个题目，还是考察的二叉树的搜索，第一印象是BFS。使用一个辅助队列存储遍历过程中的节点，遍历过程中，如果节点的值比结果中第二小的值还大，就不用继续把该节点及其所有的子节点入队了，这样可以节省时间。这里的辅助队列使用的是deque，在两端存取数据的复杂度都是O(1)，List取出第一个元素的复杂度是O(n）。代码如下：

 # Definition for a binary tree node.
 # class TreeNode(object):
 #     def __init__(self, x):
 #         self.val = x
 #         self.left = None
 #         self.right = None

 class Solution(object):
     def findSecondMinimumValue(self, root):
         """
         :type root: TreeNode
         :rtype: int
         """
         from collections import deque
         if not root:
             return -1
         stack = deque([root])
         ret = [root.val]
         while stack:
             node = stack.popleft()
             if node:
                 if node.val > ret[-1]:
                     if len(ret) < 2:
                         ret.append(node.val)
                 else:
                     if ret[0] < node.val < ret[1]:
                         ret[1] = node.val
                     stack.append(node.left)
                     stack.append(node.right)
         return ret[1] if len(ret) == 2 else -1

每次做完都要去讨论区看看有没有亮瞎眼的解法，把DFS的代码贴在下面，思路是一样的，不过看着更简洁了：

 class Solution(object):
     def findSecondMinimumValue(self, root):
         """
         :type root: TreeNode
         :rtype: int
         """
         # Time: O(n)
         # Space: O(lgn)
         res = [float('inf')]
         def traverse(node):
             if not node:
                 return
             if root.val < node.val < res[0]:
                 res[0] = node.val
             traverse(node.left)
             traverse(node.right)
         traverse(root)
         return -1 if res[0] == float('inf') else res[0]

这里需要说一下代码中的float('inf')，以前倒是真没用到这里，这是Python中代表正无穷的写法，负无穷就是float('-inf')了，又了解了一个新内容，嘿嘿。