leetcode力扣刷题打卡
题目:1201. 丑数 III
描述:给你四个整数:n 、a 、b 、c ,请你设计一个算法来找出第 n 个丑数。
丑数是可以被 a 或 b 或 c 整除的 正整数 。
解题思路
1、上来就先求最大公约数,最小公倍数;
2、容斥原理,求三个数的最小公倍数有多少个丑数;
2、二分查找;
原代码##
typedef long long ll;
ll gcd(ll a, ll b) {
return b == 0 ? a : gcd(b, a % b);
}
ll lcm(ll a, ll b) {
return a * b / gcd(a, b);
}
int nthUglyNumber(int n, int a, int b, int c) {
ll ab = lcm(a, b), ac = lcm(a, c), bc = lcm(b, c), abc = lcm(ab, c);
ll cyc = abc / a + abc / b + abc / c - abc / ab - abc / ac - abc / bc + abc / abc;
ll temp = n / cyc;
ll l = temp * abc - 1, r = (temp + 1) * abc;
while (l + 1 != r) {
ll mid = (l + r) >> 1;
ll cnt = mid / a + mid / b + mid / c - mid / ab - mid / ac - mid / bc + mid / abc;
if (cnt < n) l = mid;
else r = mid;
}
return r;
}
};
思考
1、l的初始值l = temp * abc - 1,为什么要**-1**?
2、 while (l + 1 != r),为什么要 +1 ?
2、不用上述方法进行加减一操作,而是在mid = (l + r + 1) / 2,超时,估计是死循环了,why???