leetcode 878. Nth Magical Number (第N个魔法数)

在这里插入图片描述
给出整数a, b,找到第n个魔法数。魔法数是指能被a 或 b整除。
结果对109+7取模

思路:
比如2和3,n=4,即要找第4个魔法数
魔法数能被2或3整除,先找到2和3的倍数
2 4 6 8 10 12
3 6 9 12 15
排下序:2 3 4 6 6 8 9 10 12 12 15

可以发现,每个公倍数会出现2次,以公倍数为分界点,每两个不同的公倍数之间的数字个数是一样的
把不同的公倍数之间的数字称为一个block,那么block的长度len=lcm/2 + lcm/3 - 1 ,
其中lcm是指最小公倍数,1是重复的一个公倍数,要减掉

我们要找第n个魔法数,可用n/len定位到第几个block,再用n % len定位到所在block的第几个数字
比如n=4,a=2,b=3
算出lcm = 6
block的长度len = lcm/2 + lcm/3 - 1 = 4
n/len=1, 所以我们定位在第1个block,n%len = 0, 所以第n个数在第一个block的最后一个数,也就是6

那如果n%len!=0呢?
假设shift = n%len, 即偏移量
注意这个是偏移量,即下一个block的第几个数字,而我们要找的是具体数字是多少,即要找比定位到的block的最后一个数字大了多少,大的这部分的增加量。
因此需要找偏移量和增加量的关系。

还是假设a=2,b=3
如果是2的倍数,走一步的偏移量是2, 同理3的倍数走一步的偏移量是3
一步偏移量对应的分别就是1/2,1/3
小学有道题,甲做完一个工作2小时,乙做完一个工作3小时,同时做会是几小时,那么他们一起做的工作效率是1/2+1/3
这里也是,一起移动每个偏移量对应1/2 + 1/3
那么一个偏移量对应的实际的数字增加量就是1/(1/2+1/3),也就是下一个数字比上一个数字平均大多少。当然它不能是小数,取ceil

刚刚用1/(1/2+1/3)是根据a和b一起走,根据平均效率得到的应该增加的数字,实际上只有a或b移动,分别看a和b要多少步移动到这个增加的数字,取较小值,就是实际的偏移量对应的增加量

用lcm x block数 + 增加量 即得到最后的魔法数(说的有点难理解,其实就是一个是block剩下的偏移量,一个是这个偏移量对应的增加的数,要怎么把它们对应起来的问题,对应关系如下)
偏移量 / (1/a + 1/b) = min( ceil(增加的数字/a) x a + ceil(增加的数字/b) x b)

    public int nthMagicalNumber(int n, int a, int b) {
        long lcm = a * b / gcd(a, b);
        long M = 1000000007;
        long len = lcm / a + lcm / b - 1;
        long blocks = n / len;
        long shift = n % len;
        double idx = shift / (1.0 / a + 1.0 / b);
        long reminder = Math.min((long)Math.ceil(idx/a)*a, (long)Math.ceil(idx/b) * b);
        return (int)((lcm * blocks + reminder) % M);
    }
    
    int gcd(int a, int b) {
        return (b == 0 ? a : gcd(b, a % b));
    }

还有一个二分查找法
假如第n个魔法数是num, 那么n = num/a + num/b - num/lcm
根据限制,知道left = 1, right = n × max(a,b)×109 ,设right = 1e15好了
二分查找 使mid == n即可,注意超出了整数范围,设为long型

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蓝羽飞鸟

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值