【384. 打乱数组】暴力/蓄水池算法

最新推荐文章于 2024-10-20 09:49:14 发布

如花美眷似水流年Nancy

最新推荐文章于 2024-10-20 09:49:14 发布

阅读量460

点赞数

分类专栏：保研刷题文章标签： leetcode c++ 算法

本文链接：https://blog.csdn.net/Nancy9669/article/details/123820090

版权

保研刷题专栏收录该内容

39 篇文章 1 订阅

订阅专栏

方法1：暴力

所以要用list数据结构

class Solution {
public:
    Solution(vector<int>& nums) {
        this->nums = nums;
        this->original.resize(nums.size());
        copy(nums.begin(), nums.end(), original.begin());
    }
    
    vector<int> reset() {
        copy(original.begin(), original.end(), nums.begin());
        return nums;
    }
    
    vector<int> shuffle() {
        vector<int> shuffled = vector<int>(nums.size());
        list<int> lst(nums.begin(), nums.end());
      
        for (int i = 0; i < nums.size(); ++i) {
            int j = rand()%(lst.size());
            auto it = lst.begin();
            advance(it, j);
            shuffled[i] = *it;
            lst.erase(it);
        }
        copy(shuffled.begin(), shuffled.end(), nums.begin());
        return nums;
    }
private:
    vector<int> nums;
    vector<int> original;
};

方法2：Fisher-Yates 洗牌算法

class Solution {
public:
    vector<int> nums;
    vector<int> original;
    Solution(vector<int>& nums) {
        this->nums=nums;
        this->original=nums;
    }
    
    vector<int> reset() {
        return original;
    }
    
    vector<int> shuffle() {
        int len=nums.size();
        for(int i=0;i<len;i++)
        {
            int j=i+rand()%(len-i);
            swap(nums[i],nums[j]);
        }
        return nums;
    }
};