leetcode 刷题系列 2019/4/29

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. $1<=N<=10^9$
2. $2<=A<=40000$
3. $2<=B<=40000$

解题思路：

以8与12 为例，其最小公倍数为24；在1-24内，则会有

+8，+16，+24；

+12，+24；

去除冗余部分，则每增加24，会出现（2+3）-1次数字，

故对于第N个数，其经历了N /（2+3）-1）次循环，

“整循环”部分值为：N/（2+3-1）*24

且还剩余N%(2+3-1)个数，将其排序，vc[0,8,12,16,24]

剩余部分值为 vc[N%(2+3-1)],

其和即为答案。

代码如下

class Solution:
    def nthMagicalNumber(self, N: int, A: int, B: int) -> int:
        mod=1000000007
        # 找出最大公约数 辗转相除法
        n, m = max(A,B),min(A,B)
        md=n%m
        while md:
            n, m = m, md
            md = n%m
        ma = A*B//m #最小公倍数
        vc = [0]
        for i in range(1,B//m):
            vc.append(i*A)
        for i in range(1,A//m):
            vc.append(i*B)
        vc.sort()
        c = len(vc)
        return (ma*(N//c) + vc[N%c])%mod