Randow Id generator with array of probability

最新推荐文章于 2025-06-17 15:04:27 发布

转载最新推荐文章于 2025-06-17 15:04:27 发布 · 504 阅读

Google面试专栏收录该内容

22 篇文章

订阅专栏

本文探讨了在给定多个id及其相应概率的情况下，如何通过前缀和及二分查找方法优化随机生成id的过程。重点在于提高算法效率，特别是在id数量庞大且频繁调用场景下的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给一组id和表示每个id出现概率的数组，概率之和为1.要求随机生成id，使得随机出的id满足之前的概率数组。

**如果id很多，调用这个随机生成方法的次数也很多，怎么优化

使用prefix sum，使用binary search

class RandomID {
    public static void main(String[] args) {
        int[] id = { 1, 2, 3, 4, 5 };
        double[] probility = { 0.2, 0.1, 0.4, 0.25, 0.05 };
        System.out.println(getRandomID(id, probility));
    }

    private static int getRandomID(int[] id, double[] probility) {
        int len = probility.length;
        double[] sum = new double[len];
        //the get prefix sum of the probility sum[0] = probility[0];
        for (int i = 1; i < len; i++) {
            sum[i] = sum[i - 1] + probility[i];
            System.out.println(sum[i]);
        }
        double prob = Math.random();
        System.out.println(prob);
        if (prob < sum[0]) {
            return id[0];
        }
        //binary search
        int start = 0;
        int end = len - 1;
        while (start + 1 < end) {
            int mid = start + (end - start) / 2;
            if (sum[mid] == prob) {
                end = mid;
            } else if (sum[mid] > prob) {
                end = mid;
            } else {
                start = mid;
            }
        }
        return id[start + 1];
    }
}