请你帮忙设计一个程序,用来找出第 n
个丑数。
丑数是可以被 a
或 b
或 c
整除的 正整数。
示例 1:
输入:n = 3, a = 2, b = 3, c = 5
输出:4
解释:丑数序列为 2, 3, 4, 5, 6, 8, 9, 10... 其中第 3 个是 4。
示例 2:
输入:n = 4, a = 2, b = 3, c = 4
输出:6
解释:丑数序列为 2, 3, 4, 6, 8, 9, 12... 其中第 4 个是 6。
示例 3:
输入:n = 5, a = 2, b = 11, c = 13
输出:10
解释:丑数序列为 2, 4, 6, 8, 10, 11, 12, 13... 其中第 5 个是 10。
示例 4:
输入:n = 1000000000, a = 2, b = 217983653, c = 336916467
输出:1999999984
提示:
1 <= n, a, b, c <= 10^9
1 <= a * b * c <= 10^18
[1, 2 * 10^9]
的范围内解题思路
这个问题是之前问题Leetcode 264:丑数 II(超详细的解法!!!)的提高,由于数据量的问题,所有我们要使用O(logn)
或者O(1)
的算法。由于是查找问题所有不难想到二分法。对于任意数字k
,判断小于k
的丑数有多少个?如果个数小于n
的话,我们需要增大k
,否则我们需要减少k
。
那么现在的问题就变成了判断小于k
的丑数有多少个?能被a
整除的数有k//a
个,能被b
整除的数有k//b
个,能被c
整除的数有k//c
个。但是直接将三者相加是不对的,因为有重复元素,可以根据容斥原理剔除重复元素,得到最后的解为k//a+k//b+k//c-k//lcm(a,b)-k//lcm(b,c)-k//lcm(a,c)+k//lcm(a,b,c)
(其中lcm
计算最小公倍数)。而x*y=gcd(x,y)*lcm(x,y)
(其中gcd
计算最大公约数),所以我们可以通过最大公约数计算最小公倍数。
class Solution:
def nthUglyNumber(self, n: int, a: int, b: int, c: int) -> int:
def gcd(x, y):
while y:
x, y = y, x % y
return x
def lcm2(x, y):
return x*y//gcd(x, y)
def lcm3(x, y, z):
res = x * y // gcd(x, y)
return res * z // gcd(z, res)
def cnt(k, x, y, z):
return k//x + k//y + k//z - k//lcm2(x,y) -k//lcm2(x, z) - k//lcm2(y, z) + k//lcm3(x, y, z)
l, r = 1, 2*10**9
while l < r:
mid = (l + r) >> 1
if cnt(mid, a, b, c) < n:
l = mid + 1
else:
r = mid
return l
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!