【lc刷题】230/671 二叉搜索树中第K小的元素/二叉树中第二小的节点_Day42(118-119/300)

118-119

2道题：

二叉搜索树中第K小的元素

230. 二叉搜索树中第K小的元素
     给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
      
     说明：
     你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。
      
     示例 1:
      
     输入: root = [3,1,4,null,2], k = 1

    3
   / \
  1   4
   \
    2

输出: 1
示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3

         5
        / \
       3   6
      / \
     2   4
    /
   1

输出: 3
进阶：
如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化 kthSmallest 函数？

思路：BST 二叉搜索树，证明左边都比根小，右边都比根大。
那么，中序打印出来就是一个完整的从小到大。

class Solution:
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        self.lst =[]
        def dfs(node):
            if not node: return 
            dfs(node.left)
            self.lst.append(node.val)
            dfs(node.right)
            
        dfs(root)
        return self.lst[k-1]

Stack:

class Solution:
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        stack = collections.deque(maxlen=k)
        
        while True:
            while root:
                stack.append(root)
                root = root.left
                
            root = stack.pop()
          
            if k == 1: return root.val
            
            k, root = k-1, root.right
    

二叉树中第二小的节点

671. 二叉树中第二小的节点
     给定一个非空特殊的二叉树，每个节点都是正数，并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话，那么这个节点的值不大于它的子节点的值。
     给出这样的一个二叉树，你需要输出所有节点中的第二小的值。如果第二小的值不存在的话，输出 -1 。
      
     示例 1:
     输入:

   2
  / \
 2   5
    / \
   5   7

输出: 5
说明: 最小的值是 2 ，第二小的值是 5 。
 
示例 2:
输入:

  2
 / \
2   2

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

思路：“如果一个节点有两个子节点的话，那么这个节点的值不大于它的子节点的值”， 证明跟节点是最小的。那么在dfs时，就比大小，比根大的，比之前小的。

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

class Solution:
    def findSecondMinimumValue(self, root: TreeNode) -> int:

        self.min, self.second = root.val, float('inf')
        def dfs(node):
            if not node: return
            dfs(node.left)
            if self.min < node.val < self.second:
                self.second = node.val
            dfs(node.right)
            
        dfs(root)
        
        return self.second if self.second != float('inf') else -1

so-far 二叉树类型题：
104 二叉树的最大深度
112 路径总和
113 路径总和 II
437 路径总和 III
124 二叉树中的最大路径和
49/144/145 二叉树的中序/前序/后序遍历
429/589/590 N叉树的层序/前序/后序遍历
230/671 二叉搜索树中第K小的元素/二叉树中第二小的节点