Leetcode.1201 丑数 III

最新推荐文章于 2024-09-30 14:51:11 发布

感觉画质不如…原神

最新推荐文章于 2024-09-30 14:51:11 发布

阅读量246

点赞数 1

分类专栏： Leetcode 文章标签： leetcode 算法容斥原理二分

本文链接：https://blog.csdn.net/m0_74396439/article/details/130140777

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

该问题要求找出第n个可以被a、b或c整除的正整数（丑数）。使用容斥原理计算出能被a、b、c以及它们最小公倍数整除的数的个数，然后通过二分查找找到第n个丑数。算法的时间复杂度与计算最小公倍数的复杂度相关。

摘要由CSDN通过智能技术生成

题目链接

Leetcode.1201 丑数 III Rating ： 2039

题目描述

给你四个整数：n 、a 、b 、c，请你设计一个算法来找出第 n 个丑数。

丑数是可以被 a 或 b 或 c 整除的 正整数 。

示例 1：

输入：n = 3, a = 2, b = 3, c = 5
输出：4
解释：丑数序列为 2, 3, 4, 5, 6, 8, 9, 10… 其中第 3 个是 4。

示例 2：

输入：n = 4, a = 2, b = 3, c = 4
输出：6
解释：丑数序列为 2, 3, 4, 6, 8, 9, 10, 12… 其中第 4 个是 6。

示例 3：

输入：n = 5, a = 2, b = 11, c = 13
输出：10
解释：丑数序列为 2, 4, 6, 8, 10, 11, 12, 13… 其中第 5 个是 10。

示例 4：

输入：n = 1000000000, a = 2, b = 217983653, c = 336916467
输出：1999999984

提示：

$1 <= n, a, b, c <= 10^9$
$1 <= a * b * c <= 10^18$
本题结果在 $1, 2 * 10^9]$ 的范围内

解法：容斥原理 + 二分

对于 $1, 2, 3, 4, ..., n - 1, n$ 共 $n$ 个数，能被整数 $k$ 整除的个数为 $\lfloor \frac{n}{k} \rfloor$ 。

例如， $n = 7$ ，能被 $k = 3$ 整除的个数为 $\lfloor \frac{7}{3} \rfloor = 2$ ，即 $3, 6$ 。

$l c m (a, b)$ 是 $a$ 和 $b$ 的最小公倍数。

定义 $A$ 为 $n$ 中能被 $a$ 整除的数的个数，即 $\lfloor \frac{n}{a} \rfloor$
定义 $B$ 为 $n$ 中能被 $b$ 整除的数的个数，即 $\lfloor \frac{n}{b} \rfloor$
定义 $C$ 为 $n$ 中能被 $c$ 整除的数的个数，即 $\lfloor \frac{n}{c} \rfloor$
定义 $A B$ 为 $n$ 中能被 $l c m (a, b)$ 整除的数的个数，即 $\lfloor \frac{n}{lcm(a,b)} \rfloor$
定义 $BC$ 为 $n$ 中能被 $l c m (b, c)$ 整除的数的个数，即 $\lfloor \frac{n}{lcm(b,c)} \rfloor$
定义 $A C$ 为 $n$ 中能被 $l c m (a, c)$ 整除的数的个数，即 $\lfloor \frac{n}{lcm(a,c)} \rfloor$
定义 $A BC$ 为 $n$ 中能被 $l c m (a, b, c)$ 整除的数的个数，即 $\lfloor \frac{n}{lcm(a,b,c)} \rfloor$

故 $n$ 个数中能被 a 或 b 或 c 整除的正整数个数为： $c n t = (A + B + C) - (A B + BC + A C) + (A BC)$ 。

我们用二分，找到第一个 $\geq n$ 的 $mi d$ ，即 $mi d$ 就是第 $n$ 个丑数。

时间复杂度： $l o g r$

C++代码：

using LL = long long;

class Solution {
public:
    int nthUglyNumber(int n, int a, int b, int c) {
        LL ab = lcm<LL>(a,b);
        LL bc = lcm<LL>(b,c);
        LL ac = lcm<LL>(a,c);
        LL abc = lcm<LL>(ab,c);

        LL l = 1 , r = 2e9;
        while(l < r){
            LL mid = (r - l) / 2 + l;
            LL cnt = (mid/a + mid/b + mid/c) - (mid/ab + mid/bc + mid/ac) + (mid/abc);
            if(cnt >= n) r = mid;
            else l = mid + 1;
        }

        return (int)l;
    }
};