LeetCode 打乱数组

打乱一个没有重复元素的数组。

示例:

// 以数字集合 1, 2 和 3 初始化数组。
int[] nums = {1,2,3};
Solution solution = new Solution(nums);

// 打乱数组 [1,2,3] 并返回结果。任何 [1,2,3]的排列返回的概率应该相同。
solution.shuffle();

// 重设数组到它的初始状态[1,2,3]。
solution.reset();

// 随机返回数组[1,2,3]打乱后的结果。
solution.shuffle();

思路：拿到这个题目想到的就是，这什么乱七八糟的。。。。不过仔细思考了一下题目想说的可能是对数组进行重新排列组合，并随机返回其中一个排列组合。这样的话其实就很简单了。

首先，看到reset函数的作用是重置到初始状态，所以我们需要一个数组来保存原来的数组。

其次，就是shuffle函数，打乱顺序。我们需要一个数组来进行打乱，对于数组中的每一项都有可能与其他项进行交换，至于对哪一项进行交换我们可以使用random函数来产生随机数，这就保证了等可能性。代码如下：

 

class Solution {
private:
	vector<int> tmp;
public:
	Solution(vector<int>nums) {
		tmp = nums;
	}
	vector<int> shuffle() {
		int pos, t;
		vector<int> cur;
		for (int i = 0; i < tmp.size(); i++) {
			cur.push_back(tmp[i]);
		}
		for (int i = tmp.size()-1; i >= 0; i--) {
			pos = rand() % (i+1);
			t = cur[i];
			cur[i] = cur[pos];
			cur[pos] = t;
		}
		return cur;
	}
	vector<int> reset() {
		return tmp;
	}
};

代码运行速度：

 

最快的156ms的这个代码也贴在下面，其实算法思路都是差不多的。