diff --git a/src/main/java/fr/pandacube/java/util/RandomUtil.java b/src/main/java/fr/pandacube/java/util/RandomUtil.java index a53d1bb..3a7215a 100644 --- a/src/main/java/fr/pandacube/java/util/RandomUtil.java +++ b/src/main/java/fr/pandacube/java/util/RandomUtil.java @@ -2,6 +2,7 @@ package fr.pandacube.java.util; import java.util.List; import java.util.Random; +import java.util.Set; public class RandomUtil { @@ -22,5 +23,24 @@ public class RandomUtil { public static T listElement(List arr) { return arr.get(rand.nextInt(arr.size())); } + + /** + * Returns a random value from a set. + * + * May not be optimized (Actually O(n) ) + * @param arr + * @return + */ + public static T setElement(Set set) { + if (set.isEmpty()) + throw new IllegalArgumentException("set is empty"); + int retI = rand.nextInt(set.size()), i = 0; + for (T e : set) { + if (retI == i) + return e; + i++; + } + throw new RuntimeException("Should never go to this line of code"); + } }