18.3 Write a method to randomly generate a set of m integers from an array of size n. Each element must have equal probability of being chosen.

這道題讓我們從一個(gè)數(shù)組中隨機(jī)取出m個(gè)數(shù)字，要求每個(gè)數(shù)字被取出的概率相同，其實(shí)這道題用的是之前那道18.2 Shuffle Cards的方法，同樣我們可以用遞歸和迭代兩種方法來(lái)做，遞歸的思路還用的回溯法，回溯到i+1==m的時(shí)候，取出數(shù)組中前m個(gè)數(shù)字放到結(jié)果res中，然后一層一層的返回，返回的過(guò)程中每次在[0, i]之間取出一個(gè)隨機(jī)數(shù)，然后如果這個(gè)數(shù)小于m，那么更新res[k]的值為nums[i]，這樣相當(dāng)于一種洗牌，這樣保證了每個(gè)數(shù)被取出的概率都相同，參見(jiàn)代碼如下：

解法一：

vector<int> pick(vector<int> &nums, int m, int i) {if (i + 1 < m) return {};else if (i + 1 == m) {vector<int> res(m);for (int k = 0; k < m; ++k) {res[k] = nums[k];}return res;} else {vector<int> res = pick(nums, m, i - 1);int k = rand() % (i + 1);if (k < m) res[k] = nums[i];return res;} }

當(dāng)然還有對(duì)應(yīng)的迭代的寫法，思路都一樣：

解法二：

vector<int> pick(vector<int> &nums, int m) {vector<int> res(m);for (int i = 0; i < m; ++i) {res[i] = nums[i];}for (int i = m; i < nums.size(); ++i) {int k = rand() % (i + 1);if (k < m) res[k] = nums[i];}return res; }

本文轉(zhuǎn)自博客園Grandyang的博客，原文鏈接：隨機(jī)生成數(shù)字[CareerCup] 18.3 Randomly Generate Integers ，如需轉(zhuǎn)載請(qǐng)自行聯(lián)系原博主。

總結(jié)

以上是生活随笔為你收集整理的[CareerCup] 18.3 Randomly Generate Integers 随机生成数字的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。