其他题目---从N个数中等概率打印M个数

来源:互联网 发布:大数据用什么数据库 编辑:程序博客网 时间:2024/06/03 21:58

【题目】

  给定一个长度为N且没有重复元素的数组arr和一个整数m,实现函数等概率随机打印arr中的M个数。

【要求】

  1. 相同的数不要重复打印
  2. 时间复杂度O(M),空间复杂度O(1)
  3. 可以改变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个数

  总之,就是把随机选出来的数打印出来,然后将打印的数交换到范围中的最后位置,再把范围缩小,使得被打印的数下次不可能再被选中,直到打印结束。

【代码实现】

#python3.5def printRandM(arr, m):    if arr == None or len(arr) == 0 or len(arr) < m or m < 1:        return    res = 0    n = len(arr)    while res < m:        index = int(random.random() * n)        print(arr[index], end=' ')        arr[index], arr[n-1] = arr[n-1], arr[index]        n -= 1        res += 1
原创粉丝点击