权重随机算法的java实现

一、概述

　　平时，经常会遇到权重随机算法，从不同权重的N个元素中随机选择一个，并使得总体选择结果是按照权重分布的。如广告投放、资源调度、抽奖等。

　　如有4个元素A、B、C、D，权重分别为1、2、3、4，随机结果中A:B:C:D的比例要为1:2:3:4。

　　总体思路：累加每个元素的权重A(1)-B(3)-C(6)-D(10)，则4个元素的的权重管辖区间分别为[0,1)、[1,3)、[3,6)、[6,10)。然后随机出一个[0,10)之间的随机数。落在哪个区间，则该区间之后的元素即为按权重命中的元素。

　　实现方法：

利用TreeMap，则构造出的一个树为:
　　　　B(3)
　　　 /      \
        /         \
     A(1)     D(10)
               /
             /
         C(6)

然后，利用treemap.tailMap().firstKey()即可找到目标元素。

当然，也可以利用数组+二分查找来实现。

二、源码

package com.xxx.utils;
 
import com.google.common.base.Preconditions;
import org.apache.commons.math3.util.Pair;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
 
import java.util.List;
import java.util.SortedMap;
import java.util.TreeMap;
 
 
public class WeightRandom<K,V extends Number> {
    private TreeMap<Double, K> weightMap = new TreeMap<Double, K>();
    private static final Logger logger = LoggerFactory.getLogger(WeightRandom.class);
 
    public WeightRandom(List<Pair<K, V>> list) {
        Preconditions.checkNotNull(list, "list can NOT be null!");
        for (Pair<K, V> pair : list) {
            double lastWeight = this.weightMap.size() == 0 ? 0 : this.weightMap.lastKey().doubleValue();//统一转为double
            this.weightMap.put(pair.getValue().doubleValue() + lastWeight, pair.getKey());//权重累加
        }
    }
 
    public K random() {
        double randomWeight = this.weightMap.lastKey() * Math.random();
        SortedMap<Double, K> tailMap = this.weightMap.tailMap(randomWeight, false);
        return this.weightMap.get(tailMap.firstKey());
    }
 
}

三、性能

4个元素A、B、C、D，其权重分别为1、2、3、4，运行1亿次，结果如下：

元素	命中次数	误差率
A	10004296	0.0430%
B	19991132	0.0443%
C	30000882	0.0029%
D	40003690	0.0092%

从结果，可以看出，准确率在99.95%以上。

四、额外补充对treemap.tailMap().firstKey()的解释

4.1、tailMap(K fromKey) 方法用于返回此映射，其键大于或等于fromKey的部分视图。返回的映射受此映射支持，因此改变返回映射反映在此映射中，反之亦然。

声明

以下是java.util.TreeMap.tailMap()方法的声明。

public SortedMap<K,V> tailMap(K fromKey)

参数

• fromKey--返回映射中键的低端点(包括)。

返回值

该方法调用返回此映射，其键大于或等于fromKey的部分视图。

异常

• ClassCastException--抛出此异常如果fromKey与此映射的比较器不兼容。

• NullPointerException--该异常被抛出，如果fromKey为null，并且此映射使用自然顺序，或者其比较器不允许使用null键。

• IllegalArgumentException--该异常被抛出，如果此映射本身有范围限制，并且fromKey位于范围的边界之外。

例子

下面的示例演示java.util.TreeMap.tailMap()方法的使用

package com.yiibai;

import java.util.*;

public class TreeMapDemo {
   public static void main(String[] args) {
      // creating maps 
      TreeMap<Integer, String> treemap = new TreeMap<Integer, String>();
      SortedMap<Integer, String> treemapincl = new TreeMap<Integer, String>();
            
      // populating tree map
      treemap.put(2, "two");
      treemap.put(1, "one");
      treemap.put(3, "three");
      treemap.put(6, "six");
      treemap.put(5, "five");      
      
      System.out.println("Getting tail map");
      treemapincl=treemap.tailMap(3);
      System.out.println("Tail map values: "+treemapincl);      
   }    
}

现在编译和运行上面的代码示例，将产生以下结果。

Getting tail map
Tail map values: {3=three, 5=five, 6=six}

4.2、firstKey() 方法用于返回当前这个映射第一个(最低)键。

声明

以下是java.util.TreeMap.firstKey()方法的声明。

public K firstKey()

参数

• NA

返回值

该方法调用返回在此映射第一个(最低)键。

异常

• NoSuchElementException--如果此映射为空，抛出此异常。

例子

下面的示例演示java.util.TreeMap.firstKey()方法的用法。

package com.yiibai;

import java.util.*;

public class TreeMapDemo {
   public static void main(String[] args) {
      // creating tree map 
      TreeMap<Integer, String> treemap = new TreeMap<Integer, String>();
      
      // populating tree map
      treemap.put(2, "two");
      treemap.put(1, "one");
      treemap.put(3, "three");
      treemap.put(6, "six");
      treemap.put(5, "five");
      
      System.out.println("Checking first key");
      System.out.println("First key is: "+ treemap.firstKey());
   }    
}

现在编译和运行上面的代码示例，将产生以下结果。
Checking first key
First key is: 1

 

 参考链接:https://www.cnblogs.com/waterystone/p/5708063.html

 参考链接:https://www.yiibai.com/java/util