[CareerCup] 18.3 Randomly Generate Integers 随机生成数字

简介:

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.

这道题让我们从一个数组中随机取出m个数字,要求每个数字被取出的概率相同,其实这道题用的是之前那道18.2 Shuffle Cards的方法,同样我们可以用递归和迭代两种方法来做,递归的思路还用的回溯法,回溯到i+1==m的时候,取出数组中前m个数字放到结果res中,然后一层一层的返回,返回的过程中每次在[0, i]之间取出一个随机数,然后如果这个数小于m,那么更新res[k]的值为nums[i],这样相当于一种洗牌,这样保证了每个数被取出的概率都相同,参见代码如下:

解法一:

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;
    }
}

当然还有对应的迭代的写法,思路都一样:

解法二:

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;
}

本文转自博客园Grandyang的博客,原文链接:随机生成数字[CareerCup] 18.3 Randomly Generate Integers ,如需转载请自行联系原博主。

相关文章
|
固态存储 计算机视觉 异构计算
一起来学MediaPipe(一)人脸及五官定位检测
一起来学MediaPipe(一)人脸及五官定位检测
4158 0
一起来学MediaPipe(一)人脸及五官定位检测
|
边缘计算 Cloud Native 安全
云原生技术的未来发展趋势
云原生技术的未来发展趋势
369 1
|
存储 容灾 关系型数据库
OceanBase 高可用性架构解析
【8月更文第31天】在大数据和云计算蓬勃发展的今天,数据库作为数据存储的核心组件,其稳定性和可靠性直接影响到整个系统的性能。OceanBase 是由阿里巴巴集团自主研发的一款分布式关系型数据库系统,旨在为大规模在线交易处理(OLTP)场景提供高性能、高可用性的解决方案。本文将深入探讨 OceanBase 是如何通过其独特的架构设计来确保数据的高可用性和容灾能力。
652 0
|
算法 定位技术
【逻辑设计】卡诺图 | 布尔方程式 | 最小项与最大项 | 卡诺图无关项 Don‘t cares
【逻辑设计】卡诺图 | 布尔方程式 | 最小项与最大项 | 卡诺图无关项 Don‘t cares
1014 0
|
JSON 负载均衡 网络协议
gRPC知识归档
gRPC知识归档
|
存储 关系型数据库 MySQL
MySQL8 中文参考(二十四)(1)
MySQL8 中文参考(二十四)
180 2
|
存储 API Apache
解密 parquet 文件,以及如何用 Python 去处理它(二)
解密 parquet 文件,以及如何用 Python 去处理它(二)
663 2
|
分布式计算 资源调度 负载均衡
Hadoop集群配置
【6月更文挑战第13天】
338 5
|
人工智能
《千秋诗颂》:中国文化与AI技术的完美融合
【2月更文挑战第7天】《千秋诗颂》:中国文化与AI技术的完美融合
467 3
《千秋诗颂》:中国文化与AI技术的完美融合
|
前端开发
【前端切图】CSS文字渐变和背景渐变
【前端切图】CSS文字渐变和背景渐变
207 0