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.

  • 相关阅读:
    openstack 介绍
    python实现图片批量剪裁的程序
    iOS开发多线程篇—GCD介绍
    block代码块介绍
    ios中的界面跳转方式
    NSNotificationCenter消息机制的介绍
    ios的手势操作之UIGestureRecognizer浅析
    collectionViewFlow的界面编写
    UITableView的简单应用介绍
    滚动视图创建
  • 原文地址:https://www.cnblogs.com/anxin6699/p/7048263.html
Copyright © 2011-2022 走看看