解题思路:因为丑数只能被2,3,5整除,那么只要用已有的丑数乘以2,3,5就可以得到新的丑数,但是不能依次添加靠前面的数乘以2,3,5,再添加后面的数乘以2,3,5,因为会出现前面的积比后面的积结果更大的情况,例如3 * 3 > 4 * 2
,所以用三个指针选择合适的数添加到丑数序列中,这样既可以做到每个数乘以2,3,5的积都有添加进来,又可以以正确而且不重复的顺序添加,代码如下:
class Solution:
def nthUglyNumber(self, n: int) -> int:
res = [1]
a, b, c = 0, 0, 0
for i in range(1, n):
val = min(res[a] * 2, res[b] * 3, res[c] * 5)
if res[a] * 2 == val: a += 1
if res[b] * 3 == val: b += 1
if res[c] * 5 == val: c += 1
res.append(val)
return res[-1]