剑指 Offer 49. 丑数

我们把只包含质因子 2、3 和 5 的数称作丑数（Ugly Number）。求按从小到大的顺序的第 n 个丑数。

示例:

输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:  

1 是丑数。
n 不超过1690。

解题思路：

法1：HashSet和PriorityQueue。Set数据不可重复但无序，List数据可重复但有序。使用HashSet防止插入重复数据，PriorityQueue每次取队列中最小值。每次从取出PriorityQueue取出丑数中的最小值，将其分别×2，3，5放入HashSet中，如能成功放入说明该数值不重复，可将其放入PriorityQueue中，重复操作n次即可。

class Solution {
    public int nthUglyNumber(int n) {
        int[] yinzi = {2, 3, 5};
        long chou = 0;
        PriorityQueue<Long> pri = new PriorityQueue<Long>();
        Set<Long> set = new HashSet<Long>();
        pri.offer(1L);
        set.add(1L);
        for(int i = 0; i < n; i++){
            chou = pri.poll();
            for(int j: yinzi){
                if(set.add(j * chou)){
                    pri.offer(j * chou);
                }
            }
        }
        return (int)chou;
    }
}

注意这种方法会计算很多超过了序号为n的数，其大小超出了int范围，所以注意数据转换，浪费了存储空间和计算时间。

法2：动态规划。三指针法，p2,p3,p5初始都指向1，每次计算min(p2*2, p3*3, p5*5)即为下一个丑数存入丑数数组，再判断该丑数和p2*2, p3*3, p5*5是否相等，将相等的指针后移一位。

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