丑数

描述

如果一个数只有质数因子2，3，5 ，那么这个数是一个丑数。我们可以认为 1 也是一个丑数。

样例

输入：9
输出：10

题解1: 动态规划

class Solution {
    public int nthUglyNumber(int n) {
        int a = 2, b = 3, c = 5;
        int[] dp = new int[n];
        int aa = 0, bb =0, cc=0, res;
        dp[0]=1;
        for (int i = 1; i < n; i++) {
            dp[i] = Math.min(dp[aa] * a, Math.min(dp[bb]*b, dp[cc]* c));
            if(dp[i] == dp[aa]*a) aa++;
            if(dp[i] == dp[bb]*b) bb++;
            if(dp[i] == dp[cc]*c) cc++;
        }
        return dp[n-1];
    }
}

题解2： 堆+Set

PriorityQueue: 最小堆，每次可以取出堆中当前的最小丑数

Set：检测丑数是否存在，不存在将丑数放到堆中

时间复杂度：每取出一个数，就会加入3个数，时间复杂度约等于O(3nlog(3n))

public int nthUglyNumber(int n) {
        PriorityQueue<Long> queue = new PriorityQueue<>();
        Set<Long> set = new HashSet<>();
        int[] factors = new int[]{2, 3, 5};
        queue.add(1L);
        set.add(1L);

        long min;
        for (int i = 2; i <= n; i++) {
            min = queue.poll();
            for (int factor : factors) {
                if(!set.contains(min* factor)) {
                    queue.add(min * factor);
                    set.add(min * factor);
                }
            }
        }
        min = queue.poll();
        return (int) min;
    }

标签：aa,丑数,bb,int,最小,new,动态,dp
来源： https://www.cnblogs.com/greengages/p/16330063.html