【Leetcode】878. Nth Magical Number

题目地址：

https://leetcode.com/problems/nth-magical-number/description/

给定三个正整数$n,a,b$，求第$n$个能被$a$或$b$整除的正整数。

这个数$x$即为满足$1\sim x$中有$n$个正整数能被$a$或$b$整除的最小数。显然$x$越大，$1\sim x$能被$a$或$b$整除的数的个数就越多，所以可以二分，左端点$1$，右端点 n min ⁡ { a , b } n\min\{a,b\} nmin{a,b}。而$1\sim x$能被$a$或$b$整除的数的个数可以用容斥原理，即为$\lfloor \frac{x}{a}\rfloor +\lfloor \frac{x}{b}\rfloor -\lfloor \frac{x}{(a,b)}\rfloor$，其中$(a,b)$为$a,b$的最小公倍数，其等于$\frac{ab}{\gcd (a,b)}$。代码如下：

class Solution {
 public:
  int nthMagicalNumber(int n, int a, int b) {
    const int MOD = 1e9 + 7;
    int g = gcd(a, b);
    auto check = [&](long x) { return x / a + x / b - x / (a * b / g) >= n; };
	long l = 1, r = (long)min(a, b) * n;
    while (l < r) {
      long mid = l + (r - l >> 1l);
      if (check(mid)) r = mid;
      else l = mid + 1;
    }

    return l % MOD;
  }

  int gcd(int x, int y) { return y ? gcd(y, x % y) : x; }
};

时间复杂度 O ( log ⁡ n + log ⁡ min ⁡ { a , b } ) O(\log n+\log \min \{a,b\}) O(logn+logmin{a,b})，空间 O ( log ⁡ min ⁡ { a , b } ) O(\log \min \{a,b\}) O(logmin{a,b})。