LeetCode每日一题——878. 第 N 个神奇数字

LeetCode每日一题系列

题目：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

提示：

1 <= n <= 109
2 <= a, b <= 4 * 104

思路

经典二分查找题型，具体思路看代码注释

题解

class Solution:
    def nthMagicalNumber(self, n: int, a: int, b: int) -> int:
        #a,b=的最小公倍数
        #因为后续计算的时候 需要每一个最小公倍数相当于计算了两次 需要剔除
        mcm=lcm(a,b)
        #check 函数 检查这个数是第多少个数
        def check(k:int):
            tmp=0
            tmp+=k//a
            tmp+=k//b
            tmp-=k//(mcm)
            return ans
        #二分查找 
        l=0
        r=n*max(a,b)
        while l<r:
            mid=(l+r)>>1
            cc=check(mid)
            if cc<n:
                l=mid+1
            else:
                #注意此处 如果 cc==n 这个数不一定是 结果 
                #因为如果 x是答案 x~x+(min(a,b)-1)的数的CC是一样的 
                r=mid
        #结果取模
        return r%int(1e9+7)