Python程序员面试算法宝典---解题总结: 第7章 排列组合与概率 7.5 如何以等概率地从大小为n的数组中选取m个整数

# -*- coding: utf-8 -*-

'''
Python程序员面试算法宝典---解题总结: 第7章 排列组合与概率 7.5 如何以等概率地从大小为n的数组中选取m个整数

题目:

分析:
蓄水池抽样问题。
假设已经选取了k个元素，剩余待选元素有m-k个，剩余元素有n-k个。

关键:
1 书上解法
从n个数中选择一个的概率为1/n,在剩余n-1个元素中选择一个元素的概率也是
1/n，等于第一次没选中的概率(n-1)/n * 1/(n-1)。
可以第一次选中的元素与数组中第一个元素交换，第二次从数组后面n-1选中的元素
与数组中第二个元素交换,...,直到选出m个元素，则数组前m个元素就是所求。

2 没有想到
还是之前的用法，用第一次随机选择的下标对应的元素和数组的最前面的元素交换。
最关键是利用其本身每次都是等概率的特性。
注意这里是randint(i, n-1)不是randint(0, n-1)因为之前选择的元素不能再选择了
参考:
Python程序员面试算法宝典
'''
import random

def getRandomM(array, n, m):
    if not array or n <= 0 or n < m or m <= 0:
        return
    for i in range(0, m):
        # 注意这里是randint(i, n-1)不是randint(0, n-1)因为之前选择的元素不能再选择了
        index = random.randint(i, n-1)
        array[index], array[i] = array[i], array[index]
    result = array[:m]
    return result


def process():
    array = list(range(10))
    n = len(array)
    m = n - 4
    result = getRandomM(array, n, m)
    print result


if __name__ == "__main__":
    process()