如何快速生成 1000 万个唯一数字 GroovyJava

下面的groovy代码需要大约2.5秒来生成1000万个唯一的整数

@groovy.transform.CompileStatic
Set<Integer> generateUniqueNumbers(int count, int bound=20000000, SplittableRandom rnd = new SplittableRandom()) {
    Set<Integer> data = new HashSet<Integer>(count)
    while(data.size() < count){
        data.add( rnd.nextInt(bound) )
    }
    return data
}

@groovy.transform.CompileStatic
def writeToFile(Set<Integer> data, String path){
    new File(path).withWriter("UTF-8"){w->
        data.each{i-> w.append('83').append(i.toString()).append('\n')}
    }
}


def t = System.currentTimeMillis()
def data  = generateUniqueNumbers(10000000)
println data.size()
println "generate time = ${System.currentTimeMillis() - t} millis"

//writing to file
t = System.currentTimeMillis()
writeToFile(data, "/11/out.txt")
println "write time = ${System.currentTimeMillis() - t} millis"

结果:

10000000
generate time = 2636 millis
write time = 2182 millis

如果生成数字的顺序很重要-您可以使用LinkedHashSet而不是HashSet，但这会增加~x2-x3的执行时间

注@groovy.transform.CompileStatic注释--当性能很重要并且有一个大/长循环时，请始终try 在Groovy中使用它.

UPD:使用SplittableRandom而不是Random，因为它更快

UPD2:增加了使用编写器的写入文件方法.原始问题File.append(...)中的实现每次打开和关闭文件.