LeetCode878. 第 N 个神奇数字(java)

题目

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

思路

这题我首先觉得这个模的数怎么这么奇怪，然后查了一下说，这个是int里最大的质数~那么肯定和题目无关了。

如果按最简单的思路就是一个个加上去试的话目测会超时，因为如果N，A,B都取最大值，那就是要加到40000*10^9！！

先从A = 2，B = 3找规律：
以A = 2，B = 3为例，随着N递增有 2,3,4(22),6(A,B的最小公倍数)，8(24),9(3*3)…

然后(嗯！要有点灵感)就发现 假设最后的值为 res ，此时个数 N = res/A + res/B - res/(A,B的最小公倍数)

那么当对于一个数k来说它的 个数k 如果小于N，那么res肯定在它的右边，否则在它的左边。这里我们再用2分的方式逼近res。

当k = N时，我们就命中了想要的那个数了，再对结果进行取模。

代码

import java.math.BigDecimal;
class Solution {
    public int nthMagicalNumber(int N, int A, int B) {
        /**
         * 假设最后的值为 res ，此时 N = res/A + res/B - res/(A,B的最小公倍数)
         * 以A = 2，B = 3为例，随着N递增有 2,3,4(2*2),6(A,B的最小公倍数)，8(2*4),9(3*3)...
         */
        
        int g = A*B/gcd(A,B);//最小公倍数 = A*B / 最大公约数
        
        long min = 0;
        long max = 40000000000000L;
        
        while(min < max){
            long mid = min + ((max - min) / 2);    
            long temp = mid / A + mid / B - mid / g;
            
            if(temp < N){
                min = mid + 1;
            }else {
                max = mid;
            }
        }
        return (int)(max % 1000000007);   
    }
    
    public int gcd(int a,int b){
        return b == 0 ? a : gcd(b,a%b);
    }
}