【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

求解思路

1. 这个题目感觉还是非常好的，通过二分查找+互斥原理+最大公约数+最小公倍数求解。
2. 根据题目的定义我们知道，神奇数字要么被a整除，要么被b整除。
3. 首先我们通过枚举x这个值来确定最后的结果，枚举的过程保持二段性，我们使用二分查找来优化，最终的结果来自三部分，第一部分是小于等于 x中能被 a 整除的数的个数为c1,第二部分是小于等于 x中能被 b整除的数的个数为c2，第三部分就需要结合我们的互斥原理，小于等于 x 且同时能被 a 和 b 整除的个数为c3,其中 c 为 a 和 b 的最小公倍数。最终的个数sum=c1+c2-c3，这个sum和n的值就用来控制二分查找过程的左右边界值。

实现代码

class Solution {
    static final int MOD = 1000000007;

    public int nthMagicalNumber(int n, int a, int b) {
        long l = Math.min(a, b);
        long r = (long) n * Math.min(a, b);
        int c = lcm(a, b);
        while (l <= r) {
            long mid = (l + r) / 2;
            long cnt = mid / a + mid / b - mid / c;
            if (cnt >= n) {
                r = mid - 1;
            } else {
                l = mid + 1;
            }
        }
        return (int) ((r + 1) % MOD);
    }
    //最小公倍数
    public int lcm(int a, int b) {
        return a * b / gcd(a, b);
    }
    //最大公约数
    public int gcd(int a, int b) {
        return b != 0 ? gcd(b, a % b) : a;
    }
}

运行结果