项目中需要在一个集合中获取随机记录。这个需求深入一些,发现我需要一个数组中获取随机数,还不能重复。因为数组通过下标取值,需求再次深入,我需要一个通过数组长度length与所需长度need 获取一个下标数组的方法。关键点在于length长度没有限制,于是便写了一个 RandomUtil 实现需求
-
import java.util.Arrays; import java.util.HashMap; import java.util.Map; import java.util.Random; /** * @author zhangzhaopeng * @date 2020/2/25 3:34 下午 */ public class RandomUtil { /** * 通过指定范围(0 ~ length) 和 数组数量(need),获取随机排序的int数组 * 原理:通过一个随机数选中一个元素,并与最前面的元素替换,然后从此元素往后继续挑选一个随机元 * 素,并放到前面,直到放在前面的元素数量足够,或者全部元素都得到重新排序。 * 前提条件:length >= need,如果兼容小于等于,可 need = min(lengh,need); * * @param length 可选数组范围 0 ~ length * @param need 结果数组长度 */ public static int[] getRandomArray(int length, int need){ Random random = new Random(); int[] array = new int[need]; Map<Integer, Integer> selMap = new HashMap<>(); for (int i = 0; i < need; i ++){ int r = random.nextInt(length - i); // 如果map中此元素未经过赋值,则认为它的值与key相等,然后更换元素后赋值 array[i] = selMap.getOrDefault(i + r, i + r); selMap.put(i + r, selMap.getOrDefault(i, i)); } return array; } public static void main(String[] args) { System.out.println("10 取 9:" + Arrays.toString(getRandomArray(10, 9))); System.out.println("10万 取 9:" + Arrays.toString(getRandomArray(100000, 9))); } }
返回结果:
10 取 9:[3, 4, 5, 6, 1, 7, 2, 0, 9]
10万 取 9:[24451, 76236, 29260, 61368, 54820, 87544, 54605, 79596, 17218]
Process finished with exit code 0