0264.丑数-ii.java

import java.util.HashSet;
import java.util.PriorityQueue;
import java.util.Set;

/*
 * @lc app=leetcode.cn id=264 lang=java
 *
 * [264] 丑数 II
 *
 * https://leetcode.cn/problems/ugly-number-ii/description/
 *
 * algorithms
 * Medium (58.87%)
 * Likes:    987
 * Dislikes: 0
 * Total Accepted:    142.4K
 * Total Submissions: 242K
 * Testcase Example:  '10'
 *
 * 给你一个整数 n ，请你找出并返回第 n 个 丑数 。
 * 
 * 丑数 就是只包含质因数 2、3 和/或 5 的正整数。
 * 
 * 
 * 
 * 示例 1：
 * 
 * 
 * 输入：n = 10
 * 输出：12
 * 解释：[1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 个丑数组成的序列。
 * 
 * 
 * 示例 2：
 * 
 * 
 * 输入：n = 1
 * 输出：1
 * 解释：1 通常被视为丑数。
 * 
 * 
 * 
 * 
 * 提示：
 * 
 * 
 * 1 
 * 
 * 
 */

// @lc code=start
class Solution {
    public int nthUglyNumber(int n) {
        // Set<Long> seen = new HashSet<>();
        // PriorityQueue<Long> pq = new PriorityQueue<>();
        // int[] factors = new int[] { 2, 3, 5 };

        // seen.add(1L);
        // pq.offer(1L);
        // int ugly = 0;

        // for (int i = 0; i < n; i++) {
        // long current = pq.poll();
        // ugly = (int) current;
        // for (int factor : factors) {
        // long next = current * factor;
        // if (seen.add(next))
        // pq.offer(next);
        // }
        // }

        // return ugly;

        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];
    }
}
// @lc code=end