Java实现权重随机获取值或对象的方法
作者:你就像甜甜的益达
场景
按照权重2,8给用户分组为A,B,
TreeMap.tailMap方法
treeMap是一种基于红黑树实现的有序映射表,提供了一系列的方法来操作映射表中的元素。其中tailMap方法是用于返回映射表中大于或等于给定键的部分视图。
tailMap方法的定义如下:
public NavigableMap<K,V> tailMap(K fromKey, boolean inclusive) { return new AscendingSubMap<>(this, false, fromKey, inclusive, true, null, true); }
其中,fromKey表示起始键,返回一个从fromKey开始到映射表末尾的部分视图。inclusive是表示是否包含传入的fronKey.这个部分视图是SortedMap类型的,可以进行排序操作。
使用tailMap方法需要注意以下几点:
- 如果fromKey不存在于映射表中,则返回的部分视图将包含大于fromKey的所有键值对。
- 返回的部分视图是映射表的一个视图,对这个视图所做的修改会影响到原映射表。
- 返回的部分视图是有序的,可以进行排序操作。
下面是一个示例代码:
import java.util.TreeMap; import java.util.SortedMap; public class TreeMapExample { public static void main(String[] args) { TreeMap<Integer, String> treeMap = new TreeMap<>(); treeMap.put(1, "one"); treeMap.put(2, "two"); treeMap.put(3, "three"); treeMap.put(4, "four"); treeMap.put(5, "five"); // 返回大于等于3的部分视图 SortedMap<Integer, String> tailMap = treeMap.tailMap(3); System.out.println(tailMap); // 输出 {3=three, 4=four, 5=five} // 修改部分视图 tailMap.put(6, "six"); System.out.println(treeMap); // 输出 {1=one, 2=two, 3=three, 4=four, 5=five, 6=six} // 对部分视图进行排序 SortedMap<Integer, String> sortedTailMap = tailMap.descendingMap(); System.out.println(sortedTailMap); // 输出 {6=six, 5=five, 4=four, 3=three} } }
在上面的示例代码中,首先创建了一个treeMap对象,并向其中添加了5个键值对。然后使用tailMap方法返回了大于等于3的部分视图,并对这个部分视图进行了修改和排序操作。
tailMap方法是Java中treeMap类提供的一个非常有用的方法,可以方便地获取映射表中大于等于指定键的部分视图,并进行排序和修改操作。 针对这个特性可以用来获取权重值
简单分析
好比A:B的权重为2:8,那么相当于A的权重为0->2,B的权重为2->10(2+8)都是包左不包右的; 那我们就可以随机个0-10的值,如果在0->2那么返回A,如果2->10那就返回B
使用随机值
public static String test1() { Random random = new Random(); int i = random.nextInt(10); if (i < 2) { return "A"; } else if (i >= 2 && i < 10) { return "B"; } else { return "C"; } }
使用treemap实现权重取值
public static String test2() { TreeMap<Integer, String> treeMap = new TreeMap<>(); int total = 2 + 8; treeMap.put(2, "A"); treeMap.put(total, "B"); Random random = new Random(); return treeMap.tailMap(random.nextInt(total), false).firstEntry().getValue(); }
将Int改为Double稍微准确一点,因为double随机的值更加多
public static String test3() { TreeMap<Double, String> treeMap = new TreeMap<>(); int total = 2 + 8; treeMap.put((double) 2, "A"); treeMap.put((double) total, "B"); Random random = new Random(); return treeMap.tailMap(total * random.nextDouble(), false).firstEntry().getValue(); }
测试main方法
package com.study.springbootplus.test; import cn.hutool.core.lang.WeightRandom; import cn.hutool.core.util.NumberUtil; import cn.hutool.core.util.RandomUtil; import java.util.ArrayList; import java.util.List; import java.util.Random; import java.util.TreeMap; /** * @ClassName RandomTest * @Author yida * @Date 2023-09-14 18:26 * @Description RandomTest */ public class RandomTest { public static void main(String[] args) { int num_a = 0, num_b = 0, num_c = 0; int testCount = 1000; for (int i = 0; i < testCount; i++) { switch (test3()) { case "A": num_a = num_a + 1; break; case "B": num_b = num_b + 1; break; case "C": num_c = num_c + 1; break; } } System.out.println("A-" + num_a + "-------" + NumberUtil.div(num_a, testCount, 2) * 100 + "%"); System.out.println("B-" + num_b + "-------" + NumberUtil.div(num_b, testCount, 2) * 100 + "%"); System.out.println("C-" + num_c + "-------" + NumberUtil.div(num_c, testCount, 2) * 100 + "%"); } public static String test1() { Random random = new Random(); int i = random.nextInt(10); if (i < 2) { return "A"; } else if (i >= 2 && i < 10) { return "B"; } else { return "C"; } } public static String test2() { TreeMap<Integer, String> treeMap = new TreeMap<>(); int total = 2 + 8; treeMap.put(2, "A"); treeMap.put(total, "B"); Random random = new Random(); return treeMap.tailMap(random.nextInt(total), false).firstEntry().getValue(); } public static String test3() { TreeMap<Double, String> treeMap = new TreeMap<>(); int total = 2 + 8; treeMap.put((double) 2, "A"); treeMap.put((double) total, "B"); Random random = new Random(); return treeMap.tailMap(total * random.nextDouble(), false).firstEntry().getValue(); } public static void test() { List<WeightRandom.WeightObj<String>> weightList = new ArrayList<>(); weightList.add(new WeightRandom.WeightObj<>("A", 20)); weightList.add(new WeightRandom.WeightObj<>("B", 30)); weightList.add(new WeightRandom.WeightObj<>("C", 40)); weightList.add(new WeightRandom.WeightObj<>("D", 10)); WeightRandom<String> wr = RandomUtil.weightRandom(weightList); String str = ""; int num_a = 0, num_b = 0, num_c = 0, num_d = 0; int testCount = 10000; for (int i = 0; i < testCount; i++) { str = wr.next(); switch (str) { case "A": num_a = num_a + 1; break; case "B": num_b = num_b + 1; break; case "C": num_c = num_c + 1; break; case "D": num_d = num_d + 1; break; } } System.out.println("A-" + num_a + "-------" + NumberUtil.div(num_a, testCount, 2) * 100 + "%"); System.out.println("B-" + num_b + "-------" + NumberUtil.div(num_b, testCount, 2) * 100 + "%"); System.out.println("C-" + num_c + "-------" + NumberUtil.div(num_c, testCount, 2) * 100 + "%"); System.out.println("D-" + num_d + "-------" + NumberUtil.div(num_d, testCount, 2) * 100 + "%"); } }
测试结果:
A-195-------20.0%
B-805-------81.0%
C-0-------0.0%
当权重的参数比较多,那么建议使用hutool封装的
到此这篇关于Java实现权重随机获取值或对象的方法的文章就介绍到这了,更多相关Java权重随机获取值或对象内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!