zoukankan      html  css  js  c++  java
  • [Leetcode] Heap, Divide and conquer--215. Kth Largest Element in an Array

    Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

    For example,
    Given [3,2,1,5,6,4] and k = 2, return 5.

    Solution:

     1st method using quicksort to sort in non-ascending order, and get the kth number. n = len(nums), time complexity is o(nlogn) -

     2nd method. quickselect. set the pivot (e.g. the first element ) and rearrange all the element less than pivot to be in the left, bigger than pivot to be in the right; then compare with the pivot index with k to decide to go next in the left part or right part to recursively do the same thing reference introduction to algorithm texbook or --reference https://en.wikipedia.org/wiki/Selection_algorithm https://discuss.leetcode.com/topic/14611/java-quick-select/2 random algorithm ; Amortized time complexity is o(n), worst case is o(n^2)

     1  def select(nums, l, r, index):
     2             if r == l:
     3                 return nums[l]
     4             #pivot index
     5             pind = random.randint(l, r)
     6             nums[l], nums[pind] = nums[pind], nums[l]   #move pivot to the beginning of the
     7             
     8             #partition around pivot
     9             i = l
    10             for j in xrange(l+1, r+1):
    11                 if nums[j] < nums[l]:
    12                     i += 1
    13                     nums[i], nums[j] = nums[j], nums[i]
    14             nums[i], nums[l] = nums[l],nums[i] 
    15             
    16             if index == i:
    17                 return nums[i]
    18             elif index < i:
    19                 return select(nums, l, i-1, index)
    20             else:
    21                 return select(nums, i+1, r, index)
    22             
    23         index = len(nums) - k
    24         if nums is None or len(nums) < 1:
    25             return 0
    26         return select(nums, 0, len(nums)-1, index)

     3rd method, use heap if built-in heapq is allowed in python,

    1 h = []
    2         for n in nums:
    3             heapq.heappush(h, -1*n)     #min-heap
    4             
    5         for i in range(0, k-1):
    6             heapq.heappop(h)
    7             
    8         return -1*heapq.heappop(h)

    4th - another method that I have not come up with, which is to use multiset, similar to heap ( it is interesting). --reference https://www.liuchuo.net/archives/3016.

  • 相关阅读:
    一条语句批量插入多条数据
    VMware Workstation 15 Pro 永久激活密钥
    windows下java项目打包、启动批处理 .bat文件
    java简写名词解释
    windows查看所有进程:netstat -ano
    Java验证jwt token
    【转载】三种方法从 HTTPS 网站导出 SSL 证书链
    使用solr将CSV/XML/DB/JSON数据发布为Rest Service
    检验多个xsd的xml是否合法
    Eclipse 打包运行maven项目
  • 原文地址:https://www.cnblogs.com/anxin6699/p/7048263.html
Copyright © 2011-2022 走看看