从数组中取出n个不同的随机数

有一给定数组arr(元素不重复),想要随机从中取出m个元素,怎么取?

思想:每次产生一个随机数,在arr数组中取出相应的元素,每次对选中的元素和最后元素交换以下,下一次生成随机数的时候就在前m-1个元素中生成,这样每次只要交换一次元素,就不用去那个数组中判断当前下标是否已经被生成过一次,时间复杂度变成O(m)

#从[0,m)中产生n个随机数
def nfromm(m,n,unique=True):
    if unique:
        box = [i for i in range(m)]
        ans = []
        import random
        for i in range(n):
            index = random.randint(0,m-i-1)
            
            #将选中的元素插入到输出结果列表中
            ans.append(box[index])
            
            #将选中的元素放到最后,然后在前面的元素中继续进行随机选择
            box[index],box[m-i-1]=box[m-i-1],box[index]
        return ans
    else:
        ans = []
        for i in range(n):
            ans.append(random.randint(0,m-i-1))
        return ans
nfromm(5,5)

从数组arr中取出元素也是一样的。

#从arr数组中产生n个随机数
def nfromarr(arr,n,unique=True):
    out = []
    m = len(arr)
    import random
    if unique:
        if n>m:
            return False
        for i in range(n):
            index = random.randint(0,m-i-1)
            out.append(arr[index])
            arr[index],arr[m-i-1]=arr[m-i-1],arr[index]
        return out
    else:
        for i in range(n):
            out.append(random.randint(0,m-i-1))
        return out
nfromarr([2,3,45,6,7],3)

猜你喜欢

转载自blog.csdn.net/weixin_38332967/article/details/82947460