python 中位数

 有⼀个⽆序的整数序列（⽆重复项，⻓度为奇数），请⽤你认为最优的⽅法求序列的中位数。例如给
定数组 1、5、2、9、8、0、6，中位数是 5。要求算法的时间复杂度需要⼩于 O(n2)，不能使⽤内置排
序函数，可以⽤任何语⾔实现。
输⼊：⼀个⽆序的整数序列。
输出：中位数。

顺序统计量问题

输入：一个包含n个（互异的）数的集合A和一个整数i;

输出：元素x,A中恰好有i-1个元素小于x;

方法1：时间复杂度O(nlgn)

使用堆排序或归并排序对输入数据进行排序，然后在输出数组中根据下标找出第i个元素即可

方法2：期望为线性时间的选择算法

import random
arr=[1,3,11,15,12,14,5]
def partition(arr,s,t):
    x=arr[t]
    i=s-1
    for j in range(s,t):
        if arr[j]<x:
            i+=1
            arr[i],arr[j]=arr[j],arr[i]
    arr[i+1],arr[t]=arr[t],arr[i+1]
    return i+1
def random_partition(arr,s,t):
    # print(s,t-1)
    i=random.randrange(s,t)
    arr[i],arr[t]=arr[t],arr[i]
    return partition(arr,s,t)
def random_select(arr,s,t,i):
    print(s,t,i)
    if s==t:
        return arr[s]
    q=random_partition(arr,s,t)
    k=q-s+1         #左侧分组的长度
    if k==i:
        return arr[q]
    elif k>i:
        return random_select(arr,s,q-1,i)
    else:
        return random_select(arr,q+1,t,i-k)
print(random_select(arr,0,len(arr)-1,len(arr)//2+1))