生成乘积数组的方案数

/*
 * 思路：
 * 1. 和普通Java版本类似，但尽量使用Stream来进行操作。
 * 2. 分解质因数，计算组合数，和普通解法相同。
 * 3. 通过Stream简化代码的操作。
 */

import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class SolutionStream {
    private static final int MOD = 1000000007;

    public int[] waysToFillArray(int[][] queries) {
        return Arrays.stream(queries)
            .mapToInt(query -> countWays(query[0], query[1]))
            .toArray();
    }

    private int countWays(int n, int k) {
        Map<Integer, Integer> primeFactors = getPrimeFactors(k);
        return primeFactors.values().stream()
            .mapToLong(power -> combination(n + power - 1, power))
            .reduce(1L, (a, b) -> a * b % MOD)
            .intValue();
    }

    private Map<Integer, Integer> getPrimeFactors(int num) {
        return IntStream.rangeClosed(2, (int) Math.sqrt(num))
            .filter(i -> num % i == 0)
            .collect(HashMap::new, (map, i) -> {
                while (num % i == 0) {
                    map.put(i, map.getOrDefault(i, 0) + 1);
                    num /= i;
                }
            }, HashMap::putAll);
    }

    private long combination(int n, int k) {
        return IntStream.range(0, k)
            .mapToLong(i -> (long)(n - i) * modInverse(i + 1, MOD) % MOD)
            .reduce(1L, (a, b) -> a * b % MOD);
    }

    private long modInverse(long a, int mod) {
        return power(a, mod - 2, mod);
    }

    private long power(long a, int b, int mod) {
        long res = 1;
        while (b > 0) {
            if ((b & 1) != 0) res = res * a % mod;
            a = a * a % mod;
            b >>= 1;
        }
        return res;
    }
}