一枚菜鸟的leetcode刷题笔记 - Day14

215 - 数组中的第K个最大元素

在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

class Solution:
    def findKthLargest(self, nums: List[int], k: int) -> int:
        nums.sort()  #从小到大排列
        return nums[-k]

217 - 存在重复元素

给定一个整数数组,判断是否存在重复元素。
如果存在一值在数组中出现至少两次,函数返回 true 。如果数组中每个元素都不相同,则返回 false 。

class Solution:
    def containsDuplicate(self, nums: List[int]) -> bool:
        nums.sort()
        for i in range(1, len(nums)):
            if nums[i] == nums[i-1]:
                return True
        return False

先排序,再看有没有相邻数的相等即可。

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

给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明:你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def kthSmallest(self, root: TreeNode, k: int) -> int:
        q = []
        p = root
        counter = 0
        while p or q:
            while p:
                q.append(p)
                p = p.left
            p = q.pop()
            counter += 1
            if counter == k:
                return p.val
            p = p.right

二叉搜索树是一种节点值之间具有一定数量级次序的二叉树,对于树中每个节点:

  • 若其左子树存在,则其左子树中每个节点的值都不大于该节点值;
  • 若其右子树存在,则其右子树中每个节点的值都不小于该节点值。
    一枚菜鸟的leetcode刷题笔记 - Day14_第1张图片
    若上述二叉树中序遍历,为:0 - 1 - 2 - 3 - 5 - 6 - 7 - 8 - 9

本解法实际上是用队列写dfs中的中序遍历,需要注意的是入队和出队的顺序。

你可能感兴趣的:(leetcode,二叉树,数据结构,leetcode)