[LeetCode] 384.Shuffle an Array

最新推荐文章于 2025-01-27 23:45:00 发布

原创最新推荐文章于 2025-01-27 23:45:00 发布 · 291 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

LeetCode 专栏收录该内容

14 篇文章

订阅专栏

本文深入探讨了如何通过不放回的方式随机选择数组元素，实现数组的打乱。采用等概率选择策略，确保每次选择都在剩余元素中进行，避免重复选取。通过交换操作，无需额外空间即可完成数组的随机化，提供了具体的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：打乱数组

重点是打乱数组这一块。按序给数组每个位置随机选元素，第一个在所有l个元素中等概率选，选到i1；第二个在除去i1的其他l-1个元素中等概率选，选到i2；第i个在除去已选的i-1个元素中等概率选，概率为1/(l-i-1)。
如何安排每一个元素，最开始还在想用set记录序列索引什么的，后来明白，对于第i个元素来说，前i个已经选好，从剩余元素中等概率选，这样最终概率是一样的，就是不放回的拿小球。
这样也不用开额外的空间，只要swap操作就好辣。

在这里插入图片描述

class Solution {    
public:
    vector<int> ori;
    int l;
    Solution(vector<int> nums) {
        ori=nums;
        l=nums.size();
    }
    
    /** Resets the array to its original configuration and return it. */
    vector<int> reset() {
        return ori;
    }
    
    /** Returns a random shuffling of the array. */
    vector<int> shuffle() {
        vector<int> res=ori;
        for(int i=0;i<l;i++)
        {
            int t=rand()%(l-i)+i;
            swap(res[i],res[t]);
        }
        return res;
    }
};