其他分享
首页 > 其他分享> > 丑数——最小堆;动态规划

丑数——最小堆;动态规划

作者:互联网

丑数

描述

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

image-20220531122412425

样例

输入: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