Java random实现不同权重

原创

mob649e815d65e6 2024-07-19 10:05:41 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e815d65e6的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java随机实现不同权重

在Java编程中，我们经常需要实现一些具有不同权重的随机选择。例如，在一个抽奖活动中，我们可能希望某些奖品的中奖概率更高。本文将介绍如何在Java中实现这种不同权重的随机选择，并提供代码示例。

流程图

首先，我们通过流程图来展示实现不同权重随机选择的基本步骤：

flowchart TD
    A[开始] --> B[初始化权重数组]
    B --> C[初始化随机数]
    C --> D[计算总权重]
    D --> E[生成随机数]
    E --> F[根据权重选择]
    F --> G[返回选择结果]
    G --> H[结束]

代码示例

接下来，我们将通过一个简单的Java代码示例来实现不同权重的随机选择。

import java.util.Random;

public class WeightedRandom {
    private double[] weights;
    private double totalWeight;

    public WeightedRandom(double[] weights) {
        this.weights = weights;
        this.totalWeight = calculateTotalWeight(weights);
    }

    private double calculateTotalWeight(double[] weights) {
        double total = 0;
        for (double weight : weights) {
            total += weight;
        }
        return total;
    }

    public int select() {
        Random random = new Random();
        double randomValue = random.nextDouble() * totalWeight;
        double accumulatedWeight = 0;

        for (int i = 0; i < weights.length; i++) {
            accumulatedWeight += weights[i];
            if (randomValue < accumulatedWeight) {
                return i;
            }
        }

        return -1; // 如果没有找到合适的选择，返回-1
    }

    public static void main(String[] args) {
        double[] weights = {0.2, 0.3, 0.5}; // 定义权重数组
        WeightedRandom weightedRandom = new WeightedRandom(weights);

        for (int i = 0; i < 10; i++) {
            int result = weightedRandom.select();
            System.out.println("选择结果: " + result);
        }
    }
}

甘特图

为了更好地理解不同权重随机选择的实现过程，我们可以使用甘特图来展示这个过程：

gantt
    title 权重随机选择实现过程
    dateFormat  YYYY-MM-DD
    section 初始化
    定义权重数组 :done, des1, 2024-02-01,2024-02-03
    计算总权重 :after des1, 2024-02-04,2024-02-06
    
    section 随机选择
    生成随机数 :after des2, 2024-02-07,2024-02-09
    根据权重选择 :after des3, 2024-02-10,2024-02-12
    返回选择结果 :after des4, 2024-02-13,2024-02-15