LeetCode-in-Java

1808. Maximize Number of Nice Divisors

Hard

You are given a positive integer primeFactors. You are asked to construct a positive integer n that satisfies the following conditions:

Return the number of nice divisors of n. Since that number can be too large, return it modulo 109 + 7.

Note that a prime number is a natural number greater than 1 that is not a product of two smaller natural numbers. The prime factors of a number n is a list of prime numbers such that their product equals n.

Example 1:

Input: primeFactors = 5

Output: 6

Explanation: 200 is a valid value of n. It has 5 prime factors: [2,2,2,5,5], and it has 6 nice divisors: [10,20,40,50,100,200]. There is not other value of n that has at most 5 prime factors and more nice divisors.

Example 2:

Input: primeFactors = 8

Output: 18

Constraints:

Solution

public class Solution {
    private long modPow(long b, int e, int m) {
        if (m == 1) {
            return 0;
        }
        if (e == 0 || b == 1) {
            return 1;
        }
        b %= m;
        long r = 1;
        while (e > 0) {
            if ((e & 1) == 1) {
                r = r * b % m;
            }
            e = e >> 1;
            b = b * b % m;
        }
        return r;
    }

    public int maxNiceDivisors(int pf) {
        int mod = 1000000007;
        int[] st = new int[] {0, 1, 2, 3, 4, 6};
        return pf < 5 ? pf : (int) (modPow(3, pf / 3 - 1, mod) * st[3 + pf % 3] % mod);
    }
}