力扣算法题，第N个神奇数字(Python)

878.第N个神奇数字

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

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

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

一开始是打算使用循环暴力破解

class Solution:
    def nthMagicalNumber(self, n: int, a: int, b: int) -> int:
        flag=0
        MOD=10**9+7
        for i in range(1,1000000000):
            if i%a==0:
                flag=flag+1
                if flag==n:
                    return i%MOD
            elif i%b==0:
                flag=flag+1
                if flag==n:
                    return i%MOD
            else:
                continue

结果显而易见，肯定是超出时间限制了。

后面看了官方的解析是用容斥原理+二分查找来做：
主要思路是：设 f(x)表示为小于等于 x 的「神奇数字」个数，因为小于等于 x 中能被 a 整除的数的个数为⌊x/a⌋（即x/a向下取整），小于等于 x 中能被 b 整除的个数为 ⌊x/b⌋（即x/b向下取整），小于等于 x 中同时能被 a 和 b 整除的个数为 ⌊x/c⌋（即x/c向下取整），其中 c 为 a 和 b 的最小公倍数，所以 f(x)的表达式为：

f(x) = ⌊x/a⌋+⌊x/b⌋-⌊x/c⌋

即f(x) 是一个随着 x 递增单调不减函数。那么我们可以通过「二分查找」来进行查找第 n 个「神奇数字」。

class Solution:
    def nthMagicalNumber(self, n: int, a: int, b: int) -> int:
        MOD = 10 ** 9 + 7
        l = min(a, b)#左边界值
        r = n * min(a, b)#右边界值，n*min(a,b)必包含第n个神奇数字
        c = lcm(a, b)#lcm(a,b)：取最小公倍数
        #二分查找，//：除后向下取整
        while l <= r:
            mid = (l + r) // 2
            #到中间值神奇数字的个数
            cnt = mid // a + mid // b - mid // c
            #中间值神奇数字的个数与要求的神奇数字个数作比较
            if cnt >= n:
                r = mid - 1
            else:
                l = mid + 1
        #这里也可以返回 l%MOD，一样的
        return (r + 1) % MOD
作者：力扣官方题解
链接：https://leetcode.cn/problems/nth-magical-number/solutions/1983699/di-n-ge-shen-qi-shu-zi-by-leetcode-solut-6vyy/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。