[LeetCode][Java][JavaScript]Super Ugly Number

Super Ugly Number

Write a program to find the nth super ugly number.

Super ugly numbers are positive numbers whose all prime factors are in the given prime list primes of size k. For example, [1, 2, 4, 7, 8, 13, 14, 16, 19, 26, 28, 32]is the sequence of the first 12 super ugly numbers given primes = [2, 7, 13, 19] of size 4.

Note:
(1) 1 is a super ugly number for any given primes.
(2) The given numbers in primes are in ascending order.
(3) 0 < k ≤ 100, 0 < n ≤ 106, 0 < primes[i] < 1000.

https://leetcode.com/problems/super-ugly-number/

 

 

---

 

 

和Ugly Number II一样的思路。

动态规划，ugly number肯定是之前的ugly number乘以primes中的其中一个数得到的。

结果存在dp数组中，每一个dp中的数都要乘以primes中的数，寻找所有的结果，也就是说primes中的每一个数都需要一个变量记录位置。

举例来说primes : [2, 5, 7]。

一开始2, 5, 7都指向0的位置。

每一轮循环都用2, 5, 7与当前位置的数相乘，把最小的放进dp数组中。

Java：

 

public static int nthSuperUglyNumber(int n, int[] primes) {
    int[] dp = new int[n + 1], prime = new int[primes.length];
    int dpIndex = 1, minIndex = -1;
    dp[0] = 1;
    while(dpIndex <= n){
        int min = Integer.MAX_VALUE;
        for(int i = 0; i < primes.length; i++){
            int tmp = dp[prime[i]] * primes[i];
            if(tmp < min){
                min = tmp;
                minIndex = i;
            }
        }
        prime[minIndex]++;
        if(min != dp[dpIndex - 1]){
            dp[dpIndex] = min;
            dpIndex++;
        }
    }
    return dp[n - 1];
}

 

JavaScript:

 

/**
 * @param {number} n
 * @param {number[]} primes
 * @return {number}
 */
var nthSuperUglyNumber = function(n, primes) {
    var i, dp = [1], prime = [], dpIndex = 1, tmp, min, minIndex;
    for(i = 0; i < primes.length; i++) prime[i] = 0;
    while(dpIndex <= n){
        min = Infinity;
        for(i = 0; i < primes.length; i++){
            tmp = dp[prime[i]] * primes[i];
            if(tmp < min){
                min = tmp;
                minIndex = i;
            }
        }
        prime[minIndex]++;
        if(min !== dp[dpIndex - 1]){
            dp[dpIndex] = min;
            dpIndex++;
        }
    }
    return dp[n - 1];
};