LeetCode264. Ugly Number II

Write a program to find the n-th ugly number.

Ugly numbers are positive numbers whose prime factors only include 2, 3, 5. 

Example:

Input: n = 10
Output: 12
Explanation: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 is the sequence of the first 10 ugly numbers.

---

ugly number就是公因数只能为2,3,5.所以思路就是从1开始计算，不断去乘2,3,5就可以了。比如1 -> 2, 3, 5 ... 2 -> 4, 6, 10 ... 3 -> 6, 9, 15。问题在于这样一个是有重复，一个是没有排序。

简单思路就是像上面这样分别相乘，再把所有数字set（）一下，再sort（）一下，不过这样当然效率很低。

这是在discuss里看到的方法：

相当于设了3个指针，分别代表乘以2，乘以3，乘以5，然后找这三个数的最小值。找到后记录这个数，再将指针+1，继续向下寻找。

举例说明，第一个数是1，三个指针分别是2*1,3*1,5*1，也就是2,3,5，这里最小的是2，就将2记录下来，然后指向2的指针加1，就变成了2*2，3*1，5*1，此时最小的是3，记录3以后，指向3的指针加1，就变成了2*2，3*2，5*1，以此类推。

class Solution:
    def nthUglyNumber(self, n: int) -> int:
        dp=[1]
        p2=p3=p5=0
        
        while n>1:
            num1, num2, num3=2*dp[p2], 3*dp[p3], 5*dp[p5]
            num = min(num1, num2, num3)
            
            if num==num1:p2 += 1
            if num==num2:p3 += 1
            if num==num3:p5 += 1
            dp.append(num)
            
            n-=1
        return dp[-1]