如何按步长生成min到max间的数值并避免算术溢出？

问题痛点 #

原代码在计算max - min时存在算术溢出风险，导致边缘场景下功能异常；同时需要避免低效的循环计算方式。

原实现代码 #

public static long random(long min, long max, int step) {
    if (step <= 0) {
        throw new IllegalArgumentException("step must > 0");
    }
    if (max < min) {
        throw new IllegalArgumentException("invalid range! min must <= max");
    }

    long range = (max - min) / step + 1;
    long randomIndex = ThreadLocalRandom.current().nextLong(range);
    return min + randomIndex * step;
}

正常场景示例 #

当min=1，max=10，step=2时，随机结果可为以下值之一：

1 + 2 * 0 = 1
1 + 2 * 1 = 3
1 + 2 * 2 = 5
1 + 2 * 3 = 7
1 + 2 * 4 = 9

边缘场景问题 #

当min=Long.MIN_VALUE，max=Long.MAX_VALUE，step=1时，max - min会因算术溢出得到错误结果，导致range计算异常，进而使ThreadLocalRandom.current().nextLong(range)抛出IllegalArgumentException，无法生成覆盖整个long范围的随机值。

修复后的实现 #

import java.math.BigInteger;
import java.util.concurrent.ThreadLocalRandom;

public static long random(long min, long max, int step) {
    if (step <= 0) {
        throw new IllegalArgumentException("step must > 0");
    }
    if (max < min) {
        throw new IllegalArgumentException("invalid range! min must <= max");
    }

    long range;
    try {
        // 检测溢出，无溢出则正常计算范围
        long diff = Math.subtractExact(max, min);
        range = diff / step + 1;
    } catch (ArithmeticException e) {
        // 溢出场景用BigInteger计算准确范围
        BigInteger diffBig = BigInteger.valueOf(max).subtract(BigInteger.valueOf(min));
        range = diffBig.divide(BigInteger.valueOf(step)).add(BigInteger.ONE).longValueExact();
    }

    // 覆盖全long范围时直接返回原生随机数
    if (range > Long.MAX_VALUE) {
        return ThreadLocalRandom.current().nextLong();
    }

    long randomIndex = ThreadLocalRandom.current().nextLong(range);
    return min + randomIndex * step;
}

修复说明 #

1. 溢出检测与处理：通过Math.subtractExact判断max - min是否溢出，溢出时改用BigInteger计算准确范围，避免算术错误。
2. 全范围特殊处理：当需要覆盖整个long范围时，直接调用ThreadLocalRandom.nextLong()，该方法原生支持生成Long.MIN_VALUE到Long.MAX_VALUE的随机数，无需额外计算。
3. 性能保障：仅在溢出场景下使用BigInteger（极少触发），其余场景保持原有的高效计算逻辑，避免循环带来的性能损耗。

内容的提问来源于stack exchange，提问作者Criwran