【每日一题】按权重生成随机数(二分+随机数)、不含连续1的非负整数(数位DP)、单词搜索(字典树)

528. 按权重随机选择

在这里插入图片描述

题目有点考察设计的思路的问题但是核心还是在于如何实现映射,也就是实现按照给定概率的随机分布。这里采用了类似几何概率的思想,利用前缀数组容易得到几何的概率,然后我们直需要在区间内去随机生成,就可以保证了服从给定的概率分布。

然后利用二分查找可以找到小于这个数字的最大边界。

比如[1, 3] 就会presum = [1, 4]。那当index= 3 的时候就需要返回2,其实就是target>nums[mid]时候需要left = mid+1;而等于的时候可以直接返回。

class Solution {
   
    // 使用前缀和,去解决了分段的问题。二分查找更快速的定位到对应的位置
    private static Random random = new Random();
    private int[] preSum;

    public Solution(int[] w) {
   
        this.preSum = new int[w.length];
        this.preSum[0] = w[0];
        for (int i = 1; i < w.length; i++) {
   
            this.preSum[i] = this.preSum[i - 1] + w[i];
        }
    }
    
    public int pickIndex() {
   
        int max = this.preSum[this.preSum.length - 1];
        int rand = random.nextInt(max) + 1;

        return binarySearch(preSum, rand);
    }

    private int binarySearch(int[] nums, int target) {
   
        int left = 0, right = nums.length - 1;
        while (left <= right) {
   
            int mid = (left + right) / 2;
            if (target < nums[mid]) {
   
                right = mid-1;
            } else if(target > nums[mid]){
   
                left = mid+1;
            } 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值