Leetcode528 按权重随机选择 前缀和上的二分查找

  概率的映射可以转化为前缀和上的区间映射,在前缀和中进行二分查找即可在相应概率下返回坐标。

复制代码
class Solution {

        int sum = 0;
        List<Integer> psum = new LinkedList<Integer>();
        Random random = new Random();

        public Solution(int[] w) {
            for (int num : w) {
                sum += num;
                psum.add(sum);
            }
        }

        public final int pickIndex() {
            int tar = random.nextInt(sum);
            int begin = 0, end = psum.size() - 1;
            while (begin != end) {
                int mid = (begin + end) / 2;
                if (psum.get(mid) <= tar) begin = mid + 1;
                else end = mid;
            }
            return begin;
        }
    }
复制代码

 

posted @   牛有肉  阅读(120)  评论(0编辑  收藏  举报
编辑推荐:
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· 阿里巴巴 QwQ-32B真的超越了 DeepSeek R-1吗?
· 【译】Visual Studio 中新的强大生产力特性
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 【设计模式】告别冗长if-else语句:使用策略模式优化代码结构
点击右上角即可分享
微信分享提示