LeetCode：878（Python）—— 第 N 个神奇数字（困难）

第 N 个神奇数字

概述：一个正整数如果能被 a 或 b 整除，那么它是神奇的。给定三个整数 n , a , b ，返回第 n 个神奇的数字。因为答案可能很大，所以返回答案 对 109 + 7 取模 后的值。

输入：n = 1, a = 2, b = 3
输出：2

输入：n = 4, a = 2, b = 3
输出：6

方法一：二分查找

思路：# 任何企图暴力循环破解的算法，终将超时！所以本题目难点在于二分查找。因为要求返回第 n 个神奇数字，所以需要归纳出神奇数字公式。最后用二分查找到第 n 个位置的值即可。

# 二分查找
# 任何企图暴力循环破解的算法，终将超时！
# 所以本题目难点在于二分查找。
# 因为要求返回第 n 个神奇数字，所以需要归纳出神奇数字公式。
# 最后用二分查找到第 n 个位置的值即可。
class Solution:
    def nthMagicalNumber(self, n: int, a: int, b: int) -> int:
        left = min(a, b)
        right = n * min(a, b)
        lcm = math.lcm(a, b)
        while left <= right:
            mid = (left + right) // 2
            ans = mid // a + mid // b - mid // lcm
            if ans >= n:
                right = mid - 1
            else:
                left = mid + 1
        return (right + 1) % (10 ** 9 + 7)

# 无脑暴搜超时，用双指针优化到O(n)，依然超时，直到改成二分查找，世界终于清静了

总结

无脑暴搜超时，用双指针优化到O(n)，依然超时，直到改成二分查找，世界终于清静了