java - 随机加权选择然后去除权重

Question

这个问题已经有答案了:

Random weighted selection in Java (6 个回答)

已关闭 4 年前。

我对@PeterLawrey 的评论没有任何声誉 答案:Random weighted selection in Java 。这很好。但现在我打电话next()得到结果 A，并希望从 map 中删除 A 的权重以避免再次获得A。我该如何处理？

Answer 1

您可以在 PeterLawrey 的解决方案中简单地更改此行

public E next() {
    double value = random.nextDouble() * total;
    return map.higherEntry(value).getValue();
}

至

public E next() {
    double value = random.nextDouble() * total--;
    return map.remove(higherKey(value));
}

原因remove()返回删除后关联的值

或者，如果您需要在选择后删除元素，则可以使用其他解决方案。将所有元素添加到“LinkedList”中，每个元素必须添加与其权重一样多的次数，然后使用 Collections.shuffle()，根据 JavaDoc

Randomly permutes the specified list using a default source of randomness. All permutations occur with approximately equal likelihood.

最后，使用 pop() 或 removeFirst() 获取和删除元素

Map<String, Integer> map = new HashMap<String, Integer>() {{
    put("Five", 5);
    put("Four", 4);
    put("Three", 3);
    put("Two", 2);
    put("One", 1);
}};

LinkedList<String> list = new LinkedList<>();

for (Map.Entry<String, Integer> entry : map.entrySet()) {
    for (int i = 0; i < entry.getValue(); i++) {
        list.add(entry.getKey());
    }
}

Collections.shuffle(list);

System.out.println("Size: " + list.size());
int size = list.size();
for (int i = 0; i < size; i++) {
    System.out.println(list.pop());
}

System.out.println("Size: " + list.size());