力扣 剑指 Offer 49. 丑数

力扣 剑指 Offer 49. 丑数,第1张


先给出一个比较笨的办法
通过打表的方式把所有的丑数都列出来,然后取第 n 个
所有的丑数都是 2n1 * 3n2 * 5n3 (其中n1,n2,n3 可以为0),仿照广搜的思想打表

class Solution {
    public int nthUglyNumber(int n) {
        List<Integer> l = new ArrayList<>();
        Queue<Integer> q = new LinkedList<>();
        q.offer(1);
        while(!q.isEmpty() ){
            List<Integer> t = new ArrayList<>();
            for(int i = q.size(); i > 0; i --){
                int m = q.poll();
                if(m < Integer.MAX_VALUE / 2 && m % 3 != 0 && m % 5 != 0)
                    q.offer(m * 2);
                if(m < Integer.MAX_VALUE / 3 && m % 5 != 0)
                    q.offer(m * 3);
                if(m < Integer.MAX_VALUE / 5)
                    q.offer(m * 5);
                t.add(m);
            }
            l.addAll(t);
        }
        Collections.sort(l,(o1,o2) -> o1 - o2);
        return l.get(n - 1);
    }
}

动态规划的思想
dp[n] 表示第 n 个丑数
三个指针,一个 *2 一个 *3 一个 *5,每次取较小的那个,保证dp 数组的递增

class Solution {
    public int nthUglyNumber(int n) {
        int[] dp = new int[n+1];
        dp[1] = 1;
        int p2 = 1, p3 = 1, p5 = 1;
        for (int i = 2; i <= n; i++) {
            int num2 = dp[p2] * 2, num3 = dp[p3] * 3, num5 = dp[p5] * 5;
            dp[i] = Math.min(Math.min(num2, num3), num5);
            if (dp[i] == num2) {
                p2++;
            }
            if (dp[i] == num3) {
                p3++;
            }
            if (dp[i] == num5) {
                p5++;
            }                        
        }
        return dp[n];
    }
}

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/875535.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-13
下一篇2022-05-13

发表评论

登录后才能评论

评论列表(0条)

    保存