Java根据概率(权重)排序

原理:依次将权重叠加,构建各元素命中区间,然后利用random.nextInt(int)在总的区间内生成随机整数,落在指定区间内即为命中。命中者不参与下一轮命中事件,同时命中者自身、后面的元素、以及总和均减少本轮命中值,继续下一轮,直至所有区间被命中

public class App
{
    public static void main( String[] args ) throws Exception
    {
        Channel[] channels = new Channel[4];
        channels[0] = new Channel();
        channels[0].setName("Alipay");
        channels[0].setWeight(100);
        channels[1] = new Channel();
        channels[1].setName("WxPay");
        channels[1].setWeight(90);
        channels[2] = new Channel();
        channels[2].setName("BankPay");
        channels[2].setWeight(80);
        channels[3] = new Channel();
        channels[3].setName("CashPay");
        channels[3].setWeight(70);

        Channel[] newChannels = sortWithProbability(channels);
        for (int i = 0; i < newChannels.length; i++) {
            System.out.println(newChannels[i].getName());
        }
    }

    private static Channel[] sortWithProbability(Channel[] channels) {
        Channel[] newChannels = new Channel[channels.length];
        int[] weights = new int[channels.length];
        weights[0] = channels[0].getWeight();
        for (int i = 1; i < channels.length; i++) {
            weights[i] = weights[i-1] + channels[i].getWeight();
        }
        int sum = weights[channels.length - 1];
        for (int i = 0; i < weights.length; i++) {
            System.out.print(weights[i] + " ");
        }
        System.out.println();

        Random random = new Random();
        int index = 0;
        for (int i = 0; i < channels.length; i++) {
            int weight = random.nextInt(sum);
            System.out.print("第" + i + "次: " + weight);
            int hit = 0;
            for (int j = 0; j < weights.length; j++) {
                if (weight < weights[j]) {
                    hit = j;
                    System.out.println(", 命中: " + j);
                    newChannels[index++] = channels[j];
                    sum -= channels[j].getWeight();
                    break;
                }
            }
            for (int j = hit; j < weights.length; j++) {
                weights[j] -= channels[hit].getWeight();
            }
        }
        return newChannels;
    }
}

@Data
class Channel {
    private String name;
    private int weight;
}
100 190 270 340 
第0次: 158, 命中: 1
第1次: 207, 命中: 3
第2次: 25, 命中: 0
第3次: 40, 命中: 2
WxPay
CashPay
Alipay
BankPay
上一篇 解决Eureka Server不剔除已关停的节点的问题
下一篇 分布式Snowflake雪花算法
目录
文章列表
1 OpenCart Twig模板引擎添加evaluate过滤器
OpenCart Twig模板引擎添加evaluate过滤器
2
获取select被选中option的value和text
获取select被选中option的value和text
3
高中数学基础:函数的极限及自然常数e的由来
高中数学基础:函数的极限及自然常数e的由来
4
Layer For Mobile 移动端弹出层
Layer For Mobile 移动端弹出层
5
Groovy代码示例 - 如何为groovy脚本编译后的class文件指定名字?
Groovy代码示例 - 如何为groovy脚本编译后的class文件指定名字?
最新评论
一位WordPress评论者
一位WordPress评论者
2月12日
您好,这是一条评论。若需要审核、编辑或删除评论,请访问仪表盘的评论界面。评论者头像来自 Gravatar。