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);
    }
1
2
3
4
5

其中，fromKey表示起始键，返回一个从fromKey开始到映射表末尾的部分视图。inclusive是表示是否包含传入的fronKey.这个部分视图是SortedMap类型的，可以进行排序操作。

使用tailMap方法需要注意以下几点：

1. 如果fromKey不存在于映射表中，则返回的部分视图将包含大于fromKey的所有键值对。
2. 返回的部分视图是映射表的一个视图，对这个视图所做的修改会影响到原映射表。
3. 返回的部分视图是有序的，可以进行排序操作。

下面是一个示例代码：

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}
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

在上面的示例代码中，首先创建了一个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";
        }
    }
1
2
3
4
5
6
7
8
9
10
11

使用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();
    }
1
2
3
4
5
6
7
8

将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();
    }

1
2
3
4
5
6
7
8
9

测试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 + "%");
    }


}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

测试结果:
A-195-------20.0%
B-805-------81.0%
C-0-------0.0%
如果测试基数越大,则越准确

当权重的参数比较多,那么建议使用hutool封装的

使用hutool返回权重值 原创