热线电话:13121318867

登录
首页精彩阅读Python实现返回数组中第i小元素的方法示例
Python实现返回数组中第i小元素的方法示例
2018-03-27
收藏

Python实现返回数组中第i小元素的方法示例

这篇文章主要介绍了Python实现返回数组中第i小元素的方法,结合实例形式分析了Python针对数组的遍历、排序、运算等相关操作技巧.

分享给大家供大家参考,具体如下:

#! /usr/bin/env python
#coding=utf-8
#期望为线性时间的选择算法
importrandom
classRandomSelect(object):
  defPartition(self,a, p, r):
    x=a[r]
    i=p-1
    forjinrange(p, r):
      '''如果a[j]>x,则只需将j的值加1即可使循环不变量继续保持;
      如果a[j]<=x,则将下标i的值加1,并交换a[i]和a[j],再将
      j的值加1.此时循环不变量同样得到保持'''
      ifa[j]<=x:
        i=i+1
        a[i], a[j]=a[j], a[i]
    a[i+1], a[r]=a[r], a[i+1]
    returni+1
  defRandomPartition(self,a, p, r):
    i=random.randint(p, r)#生成的随机数为p=<i<=r
    a[r], a[i]=a[i], a[r]
    returnself.Partition(a, p, r)
  defrandomSelect(self,a,p,r,i):
    ifp==r:
      returna[p]
    q=self.RandomPartition(a,p,r)
    k=q-p+1
    ifi==k:
      returna[q]
    elifi<k:
      returnself.randomSelect(a,p,q-1,i)
    else:
      returnself.randomSelect(a,q+1,r,i-k)
if__name__=='__main__':
  print"脚本之家测试结果:"
  a=[random.randint(0,20)foriinrange(10)]
  printa
  #a=sorted(a)
  #print a
  r=RandomSelect()
  r.randomSelect(a,0,len(a)-1,3)
  printa[2]#数组中的第三小的数
  a=sorted(a)
  printa

运行结果:

数据分析咨询请扫描二维码

最新资讯
更多
客服在线
立即咨询