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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/nth-magical-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Java

class Solution {
    public int nthMagicalNumber(int n, int a, int b) {
        //在区间 [1,N] 一共有 N / a + N / b - N / lcm(a,b) 个可以整除a或b的数
        int mod = (int)Math.pow(10,9) + 7;
        long lcm = a / gcd(a,b) * b;
        long left = 0;
        long right = Long.MAX_VALUE;
        while(left <= right){
            long mid = (left + right) >> 1;
            if(mid / a + mid / b - mid / lcm < n){
                left = mid + 1;
            }else {
                right = mid - 1;
            }
        }    
        return (int)(left % mod);
    }
    public int gcd(int a,int b){
        if(a % b == 0){
            return b;
        }
        return gcd(b,a % b);
    }
}