有一给定数组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)