码迷,mamicode.com
首页 > 编程语言 > 详细

java 加权随机数实现

时间:2015-04-28 18:23:50      阅读:150      评论:0      收藏:0      [点我收藏+]

标签:


import java.util.HashMap;
import java.util.Map;
import java.util.Random;

public class Test1 {

    // String 可以为任意类型 也可以自定义类型
    static Map<String, Integer> keyChanceMap = new HashMap<String, Integer>();
    static {
//        keyChanceMap.put("出现比例为10的", 10);
        keyChanceMap.put("出现比例为40的", 4);
        keyChanceMap.put("出现比例为20的", 2);
        keyChanceMap.put("出现比例为10的", 1);
        keyChanceMap.put("出现比例为30的", 3);
    }

    public static void main(String[] args) {
        Map<String, Integer> count = new HashMap<String, Integer>();
        for (int i = 0; i < 100000; i++) {

            String item = chanceSelect(keyChanceMap);

            if (count.containsKey(item)) {
                count.put(item, count.get(item) + 1);
            } else {
                count.put(item, 1);
            }
        }

        for (String id : count.keySet()) {
            System.out.println(id + "\t出现了 " + count.get(id) + " 次");
        }
    }

    public static String chanceSelect(Map<String, Integer> keyChanceMap) {
        if (keyChanceMap == null || keyChanceMap.size() == 0)
            return null;

        Integer sum = 0;
        for (Integer value : keyChanceMap.values()) {
            sum += value;
        }
        // 从1开始
        Integer rand = new Random().nextInt(sum) + 1;

        for (Map.Entry<String, Integer> entry : keyChanceMap.entrySet()) {
            rand =rand- entry.getValue();
            // 选中
            if (rand <= 0) {
                String item = entry.getKey();
                return item;
            }
        }

        return null;
    }
}

java 加权随机数实现

标签:

原文地址:http://blog.csdn.net/tiantangpw/article/details/45339413

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!