java实现权重随机算法

权重随机算法在抽奖,资源调度等系统中应用还是比较广泛的,一个简单的按照权重来随机的实现,权重为几个随机对象(分类)的命中的比例,权重设置越高命中越容易,之和可以不等于100;

简单实现代码如下:

    import java.util.ArrayList;  
    import java.util.List;  
    import java.util.Random;  
      
    public class WeightRandom {  
        static List<WeightCategory>  categorys = new ArrayList<WeightCategory>();  
        private static Random random = new Random();  
          
        public static void initData() {  
            WeightCategory wc1 = new WeightCategory("A",60);  
            WeightCategory wc2 = new WeightCategory("B",20);  
            WeightCategory wc3 = new WeightCategory("C",20);  
            categorys.add(wc1);  
            categorys.add(wc2);  
            categorys.add(wc3);  
        }  
      
        public static void main(String[] args) {  
              initData();  
              Integer weightSum = 0;  
              for (WeightCategory wc : categorys) {  
                  weightSum += wc.getWeight();  
              }  
      
              if (weightSum <= 0) {  
               System.err.println("Error: weightSum=" + weightSum.toString());  
               return;  
              }  
              Integer n = random.nextInt(weightSum); // n in [0, weightSum)  
              Integer m = 0;  
              for (WeightCategory wc : categorys) {  
                   if (m <= n && n < m + wc.getWeight()) {  
                     System.out.println("This Random Category is "+wc.getCategory());  
                     break;  
                   }  
                   m += wc.getWeight();  
              }  
      
                
        }  
      
    }  
      
    class WeightCategory {  
        private String category;  
        private Integer weight;  
          
      
        public WeightCategory() {  
            super();  
        }  
      
        public WeightCategory(String category, Integer weight) {  
            super();  
            this.setCategory(category);  
            this.setWeight(weight);  
        }  
      
      
        public Integer getWeight() {  
            return weight;  
        }  
      
        public void setWeight(Integer weight) {  
            this.weight = weight;  
        }  
      
        public String getCategory() {  
            return category;  
        }  
      
        public void setCategory(String category) {  
            this.category = category;  
        }  
    }  

 

posted @ 2017-06-30 21:33  牧之丨  阅读(9136)  评论(0编辑  收藏  举报