从N个数中等概率打印M个数
程序员文章站
2024-02-27 14:53:33
...
题目:给定一个长度为N且没有重复元素的数组arr和一个整数m,实现函数等概率随机打印arr中的M个数。
要求:
相同的数不要重复打印
时间复杂度O(M),空间复杂度O(1)
可以改变arr数组
基本思路
具体过程如下:
1、在arr[0…N-1]中随机得到一个位置a,然后打印arr[a]。
2、把arr[a]和arr[N-1]交换。
3、在arr[0…N-2]中随机得到一个位置b,然后打印arr[b],因为打印过的arr[a]已经被换到了N-1位置,所以这次打印不可能在出现。
4、把arr[b]和arr[N-2]交换。
5、在arr[0…N-3]中随机得到一个位置c,然后打印arr[c],因为打印过的arr[a]和arr[b]已经被换到了N-1位置和N-2位置,所以这次打印不可能在出现。
6、依次类推直到打印M个数
总之,就是把随机选出来的数打印出来,然后将打印的数交换到范围中的最后位置,再把范围缩小,使得被打印的数下次不可能再被选中,直到打印结束
def printRandomM(arr,m):
if arr == None or len(arr) < 1 or m < 1 or len(arr) < m:
return
length = len(arr)
res = 0
while res < m:
index = int(random.random()*n)
print(arr[index])
arr[index],arr[n-1] = arr[n-1],arr[index]
n-=1
res +=1
下一篇: 字符串中数字子串的求和