LeetCode 878. 第 N 个神奇数字--二分查找

878. 第 N 个神奇数字

如果正整数可以被 A 或 B 整除，那么它是神奇的。

返回第 N 个神奇数字。由于答案可能非常大，返回它模 10^9 + 7 的结果。

示例 1：

输入：N = 1, A = 2, B = 3
输出：2

示例 2：

输入：N = 4, A = 2, B = 3
输出：6

示例 3：

输入：N = 5, A = 2, B = 4
输出：10

示例 4：

输入：N = 3, A = 6, B = 4
输出：8

提示：

1 <= N <= 10^9
2 <= A <= 40000
2 <= B <= 40000

通过次数2,946
提交次数11,4

题解：

如果给定数字x,然后求问数字x是A和B的第多少个神奇数字，怎么算呢？
很简单吧，首先数字x中有x/A个关于A的倍数，有x/B个关于B的倍数，但是其中可能有关于A和B的重复的数字，于是得到A和B的最小公倍数base，减去x/base。
所以给定数字x，为A和B的第k个神奇的数字，k=x/A+x/B-x/base;
于是我们二分查找这个x。

AC代码

class Solution {
public:
    typedef long long ll;
    ll mod=1e9+7;
    ll get_num(int A,int B,ll x)
    {
        ll t=__gcd(A,B);
        ll base=(A/t)*B;
        return x/A+x/B-x/base;
    }
    int nthMagicalNumber(int N, int A, int B) {
        ll l=2,r=1e18,fin;
        while(l<=r)
        {
            ll mid=(l+r)/2;
            if(get_num(A,B,mid)>=N)
            {
                r=mid-1;
                fin=mid;
            }
            else l=mid+1;
        }
        ll res=fin%mod;
        return int(res);
    }
};