数组的RandomShuffle算法

试题：

random(a,b)是可以产生[a,b]之间一个随机整数的函数。请使用random(a,b)写一个算法将长度为n的整型数组随机打乱。

void random_shuffle(int a[], int n);

 

分析：

不考虑数组中元素是否重复，长度为n的数组，其全排列共有n!种。要做到概率上的随机打乱，打乱后的排列的概率需要是1/n!。

 

答案一、《算法导论》中的做法：

void random_shuffle(int a[], int n)
{
    int temp_index, temp_val;

    for (int i = 0; i < n; i++)
    {
        temp_index = random(i, n-1);

        temp_val = a[temp_index];
        a[temp_index] = a[i];
        a[i] = temp_val;
    }
} 

 

证明：

（证明时，使用1-n的下标,注意与实现区分）。

使用如下的循环不等式：在for循环的第k次迭代之后，对每个可能的k排列，子数组a[1,k]包含这个k排列的概率是(n-k)!/n!

 初始：当k=1时，第一个元素的概率是1/n，而循环不等式(n-1)!/n!=1/n。循环不等式成立。

 保持：假设当k=i时，循环不等式也成立。则当k=i+1时，k排列的概率=(k-1)排列的概率 * 第k个元素的概率。

((n-i)!/n!) * (1/(n-i)) = (n-i-1)!/n! = (n-(i+1))!/n!

循环不等式成立。 

 终止：当k=n时，((n-(n-1))!/n!) * 1 = 1/n!

 由此证明算法random_shuffle产生的数组排列是均匀随机的。

更多内容可参考《算法导论》引理5.5

 

答案二： VC STL中algorithm的做法：

void random_shuffle(int a[], int n)
{
    int temp_index, temp_val;

    // 注意这里i是从1开始，i=0时不需要交换
    for (int i = 1; i < n; i++)
    {
        temp_index = random(0, i);

        temp_val = a[temp_index];
        a[temp_index] = a[i];
        a[i] = temp_val;
    }
}

 

证明：

（证明时，使用1-n的下标,注意与实现区分）。

使用如下的循环不等式：在for循环的第k次迭代之后，对每个可能的k排列，子数组a[1,k]包含这个k排列的概率是1/k!

 初始：当k=2时，第一个元素的概率是1/2，而循环不等式1/k!=1/2。循环不等式成立。

 保持：假设当k=i时，循环不等式也成立。则当k=i+1时，k排列的概率=(k-1)排列的概率 * 第k个元素的概率。

(1/i!) * (1/(i+1)) = 1/(i+1)!

循环不等式成立。

 终止：当k=n时，(1/(n-1)!) * (1/n) = 1/n!

 由此证明算法random_shuffle产生的数组排列是均匀随机的。

 

小结：

《算法导论》是本很经典的书，给出了均匀排列的证明。而vc的stl算法给出了另外一种实现。对比之下，stl的算法更工程一些，因为i是1开始，而不是从0开始，当i=0时，random(0,0)还是自己，不需要交换，stl直接省却了这一步。同样《算法导论》的实现，for循环中的i<n，也可以改为i<n-1。

 

转载于:https://www.cnblogs.com/shokey520/p/3806628.html